首页 > 其他 > 详细

Pandas一个需求:存在一个列表,需要在一个DataFrame中取到以该列表为索引的数据

时间:2021-05-24 09:35:18      阅读:18      评论:0      收藏:0      [点我收藏+]

 需求:Pandas一个需求:存在一个列表,需要在一个DataFrame中取到以该列表为索引的数据

这里有一个坑,

In [103]: s = pd.Series([1, 2, 3])

In [104]: s
Out[104]: 
0    1
1    2
2    3
dtype: int64

当loc[]中的列表包含于S的索引中的话,没有问题

In [105]: s.loc[[1, 2]]
Out[105]: 
1    2
2    3
dtype: int64


In [4]: s.loc[[1, 2, 3]]

Passing list-likes to .loc with any non-matching elements will raise
KeyError in the future, you can use .reindex() as an alternative.

See the documentation here:
https://pandas.pydata.org/pandas-docs/stable/indexing.html#deprecate-loc-reindex-listlike

 由于 3不在s的索引中,所以引发了一个错误。

 

解决办法:

将s先用reindex() 接收需索引的列表, 这样就会将s索引中没有的值,填充为Nan

In [106]: s.reindex([1, 2, 3])
Out[106]: 
1    2.0
2    3.0
3    NaN
dtype: float64

另外:如果你只想取到给定list和DataFrame索引中交集的值,那么就可以用一下方式:
In [107]: labels = [1, 2, 3]

In [108]: s.loc[s.index.intersection(labels)]
Out[108]: 
1    2
2    3
dtype: int64


同时需要注意:
技术分享图片

 

 

 
 

Pandas一个需求:存在一个列表,需要在一个DataFrame中取到以该列表为索引的数据

原文:https://www.cnblogs.com/yc3110/p/14802890.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!