Question

我正在尝试读取CSV文件，执行一些过滤并将其另存为文本文档

Item Code Company
A    AAA   X
B    BBB   X
C    CCC   Y
D    DDD   Y
E    EEE   Y

我编写了一个程序来对项目进行分类并按公司编码，并将其保存到相应的txt文件中

a=list(set(df['Company']))
print(a)

for i in a:
    m=df.loc[df['Company'] == i]
    print(m)
    df= pd.DataFrame(m)
    df.to_csv(i+".txt", index=False)

我的输出是

['X', 'Y', 'Z']
   Item  Code Company
0    A    AAA    X
1    B    BBB     X
Empty DataFrame
Columns: [Item, Code, Company]
Index: []
Empty DataFrame
Columns: [Item, Code, Company]
Index: []

我有3个txt输出，但是只有X.txt有数据，Y和Z为空。知道我缺少什么吗？

Answer 1

那是因为您要覆盖循环中的df变量

Answer 2

这是我的错误，我在循环中修改了df val。我将代码更改为

df1=df
a=df['Company'].unique()
print(a)

for i in a:
    print(df1.loc[df['Company'] == i])
    m=df1.loc[df['Company'] == i]
    print(m)
    df1= pd.DataFrame(m)
    df1.to_csv(i+".txt", index=False)
    df1=df

成功了

熊猫数据框错误-数据框为空消息

2 个答案: