在熊猫中应用条件后如何将结果保存到新的数据框中?

时间:2020-12-21 14:12:47

标签: python pandas

我的 ID 用以下行标识:

inactiveOne =  df['id'].loc[(df['Button'] == '-') & (df['True/False'] == 'TRUE') & (df['status'] == 'OK')]

如何将我的 ID 保存为新数据框中的新列?

当我执行以下操作时,它会创建一个文件,但我将所有行都截断到一个单元格中,而不是有数百行带有我的 ID。

df2 = pd.DataFrame(data={"id": [inactiveOne]})
df2.to_csv('test.csv', index=False)

2 个答案:

答案 0 :(得分:0)

inactiveOne 周围不需要额外的括号,只需:

df2 = pd.DataFrame(data={"id": inactiveOne})
df2.to_csv('test.csv', index=False)

答案 1 :(得分:0)

pandas.loc 正如您所写的那样,它将返回一个支持 to_csv() 方法的系列对象。无需先创建另一个数据框对象。你可以简单地做

inactiveOne =  df['id'].loc[(df['Button'] == '-') & (df['True/False'] == 'TRUE') & (df['status'] == 'OK')]
inactiveOne.to_csv('test.csv',index=False)

相关问题