簡単な csv ファイルの読み書き操作

1.csvファイルを書く

import csv

'''写入csv文件'''
# encoding：编码格式，  newline=""可让数据不用隔行写入
with open('demo.csv','w',encoding='utf-8',newline="",) as datacsv:
    csv_writer = csv.writer(datacsv, dialect=("excel"),)
    csv_writer.writerow(['A','B','C','D','E','F'])
    csv_writer.writerow(['1','2','3','4','5','6'])

結果：

2.csvファイルの読み込み

'''读csv文件'''

f = open('demo.csv', 'r')
reader = csv.reader(f)
# print(next(reader))  # 读取一行
# print(next(reader))  # 读取下一行

# for循环读取
for i in reader:
    print(i)

結果：

3. panda ライブラリのファイルの読み取りと書き込み (より簡単)

'''pandas库读写操作'''

import pandas as pd
filename = 'demo.csv'
# 读文件
df = pd.read_csv(filename)
print(df)

# 取出df的前5行
data = df.head()
# 写入文件
data.to_csv('demo_01.csv')

結果：

pandas の書き込み操作にはもう少しあることがわかりました。1 行目のコンマと 2 行目の 0 (下にデータがある場合は、1、2、3 が必要です...)、これらは次のとおりです。インデックス インデックスと呼ばれる、csv ファイル内のインデックス インデックスを削除するには、書き込み時に index=False の文を追加するだけです。

data.to_csv('demo_01.csv',index=False)

結果：