如何将数据框列保存为列表? [熊猫]

时间:2016-02-16 20:16:23

标签: python pandas

我有一个DataFrame,我想在文本文件中将列保存为列表,到目前为止,我这样做:

with open('file_name', 'w') as output:    
    output.write(str(df_phone_numbers['column']))

我在这里的问题是我已经

使用pandas处理路径

filepath_or_buffer='some_path' 
path_or_buf='some_output_path'

我宁愿保持这方面的一致性。

EDIT。

我不希望此文件为csv。

我目前正在将DataFrame保存为csv

df.to_csv(path_or_buf='home/.../df.csv')

但我想要这个特定的列,作为一个txt文件,其格式使用上面的脚本保存但使用pandas而不是:

  

打开

功能

2 个答案:

答案 0 :(得分:2)

此代码与您的代码具有相同的结果。

>>>df = pd.DataFrame({'column1':[1,3],'column2':[4,5]})

>>>df['column1'].to_csv('file_name.txt',sep=' ')
>>>cat file_name.txt
0 1
1 3

将列保存为file_name.txt的其他各种方法包括:

>>>df.to_csv('file_name.txt',columns=['column1'],index=False) 
>>>cat file_name.txt
column1
a
b 

>>>df['column1'].to_csv('file_name.txt',index=False)
>>>cat file_name.txt
a
b

>>>df.to_csv('file_name.txt',columns=['column1'],index=False,line_terminator=',')
>>>cat file_name.txt
column1,a,b,

答案 1 :(得分:1)

df = pd.DataFrame({'phone_number': ['555-111-2222', '555-111-3333']})

df.phone_number.to_csv('phone_numbers.csv')

如果你不想索引:

df.phone_number.to_csv('phone_numbers.csv', index=False)

如果你想要一个txt文件,只需更改文件名即可。结果是一样的。

df.phone_number.to_csv('phone_numbers.txt', index=False)