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

 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

猜你喜欢

转载自www.cnblogs.com/medik/p/11074985.html