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')
결과:
팬더의 쓰기 작업에 조금 더 있음을 발견했습니다. 첫 번째 줄의 쉼표와 두 번째 줄의 0(아래에 데이터가 있는 경우 1, 2, 3...이 있어야 합니다.) csv 파일에서 색인 색인을 제거하려면 작성 시 index=False 문장을 추가하면 됩니다.
data.to_csv('demo_01.csv',index=False)
결과: