gpt4 book ai didi

python - 从另一个 DataFrame 的索引列表中提取 DataFrame

转载 作者:行者123 更新时间:2023-12-04 01:52:11 31 4
gpt4 key购买 nike

我有一个数据框“A”和一个索引列表“I”。我想生成/获取一个数据帧“B”,它只包含原始数据帧“A”的那些索引“I”中的数据。我怎样才能做到这一点?

假设 I = [1, 3] ,我试过这个 A.filter(items=I, axis=0) 这是正确的方法吗?一个更好的方法来做到这一点。

最佳答案

我认为需要DataFrame.loc :

A = pd.DataFrame({
'A': ['a','a','a','a','b','b','b','c','d'],
'B': list(range(9))
})
print (A)
A B
0 a 0
1 a 1
2 a 2
3 a 3
4 b 4
5 b 5
6 b 6
7 c 7

I = [1,3]
B = A.loc[I]
print (B)
A B
1 a 1
3 a 3

关于python - 从另一个 DataFrame 的索引列表中提取 DataFrame,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/52492928/

31 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com