gpt4 book ai didi

python - 获取给定条件成立的数据帧的列号

转载 作者:行者123 更新时间:2023-12-01 02:21:12 25 4
gpt4 key购买 nike

我有以下代码:

raw_data = [[1, 2, 3], [4, 5, 6]]
df = pd.DataFrame(data=raw_data,
columns=["cA", "cB", "cC"])

wrong_indexes = df.loc[df['cA'] > 2 ]
print(wrong_indexes)

打印:

   cA  cB  cC
1 4 5 6

与此相反,我只想获取满足此条件的索引列表,如下所示:

[1]

知道我该怎么做吗?

最佳答案

wrong_indexes = df.loc[df['cA'] > 2 ].index.tolist()

关于python - 获取给定条件成立的数据帧的列号,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/47932605/

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