python用pandas工具包来处理.csv文件，包括读写和切片

Posted 2022-02-06 medik

　csv是Comma-Separated Values的缩写，是用文本文件形式储存的表格数据。我们用python处理.csv文件时，发现用pandas工具包比csv工具包要方便很多，下面介绍一下一些基本的操作，如读写(read, write)和切片(slice)。
　　
　　写(write)操作：

import pandas as pd

# 每个列表表示csv文件里面的一列
a = [1, 2, 3]
b = [4, 5, 6]
c = [7, 8, 9]

# 字典中的key值就是csv中的列名
csv_file = pd.DataFrame(‘x‘: a, ‘y‘: b, ‘z‘: c)

# 把csv_file存为test.csv，index表示是否显示行名，default=True
csv_file.to_csv("test.csv", index=False)

　　下面显示的是test.csv文件的内容：

x,y,z
1,4,7
2,5,8
3,6,9

　　读(read)操作更简单：

rows = pd.read_csv(‘test.csv‘)
print(rows)
这是读出来的rows数据：

  x   y  z

0 1 4 7
1 2 5 8
2 3 6 9

下面是切片(slice)操作：

row_slice = rows.iloc[0:2, 1:3] # 取rows的0-1行，1-2列的元素

print(row_slice)

row_slice数据为：

   y z
0 4 7
1 5 8