Python数据分析基础

CSV

文件

｜

67

output

来保存你要在每一行中保留的值。第

19

行代码中的

for

循环在

my_columns_index

中的索引值之间迭代，第

20

行代码将数据行中具有这些索引值的值加入

row_list_output

。

最后，第

21

行代码将

row_list_output

中的值写入输出文件。

在命令行中运行以下脚本：

python 7csv_reader_column_by_name.py supplier_data.csv output_files\7output.csv

你可以打开输出文件

7output.csv

查看一下结果。

2. pandas

要使用

pandas

根据列标题选取列，在文本编辑器中输入下列代码，然后将文件保存为

pandas_column_by_name.py

（这个脚本读取

CSV

文件，将发票号码列与购买日期列打印到

屏幕，并将同样的列写入输出文件）：

#!/usr/bin/env python3

import

pandas

as

pd

import

sys

input_file = sys.argv[1]

output_file = sys.argv[2]

data_frame = pd.read_csv(input_file)

data_frame_column_by_name = data_frame.loc[:, ['Invoice Number', 'Purchase Date']]

data_frame_column_by_name.to_csv(output_file, ...