批量导入CSV文件数据(初级篇)
先导入需要读取得文件夹‘成都链家二手房数据’
初级篇
import csv
name_list = [
'双流', '大邑', '崇州', '彭州', '成华', '天府新区', '天府新区南区', '新津',
'新都', '武侯', '温江', '简阳', '蒲江', '邛崃', '郫都', '都江堰',
'金堂', '金牛', '锦江', '青白江', '青羊', '高新', '高新西', '龙泉驿'
]
file = open('./成都链家二手房.csv', 'w', encoding='utf-8', newline='')
mywrite = csv.writer(file)
mywrite.writerow(['行政区', '标题', '小区', '街道', '户型', '面积', '装修', '单价', '总价'])
for i in name_list:
f = open(f'./成都链家二手房数据/成都{i}二手房信息.csv', 'r', encoding='utf-8')
read_data = csv.reader(f)
for i in read_data:
if i != ['行政区', '标题', '小区', '街道', '户型', '面积', '装修', '单价', '总价']:
mywrite.writerow(i)
f.close()
file.close()
print('写入完成')
运行结果
高级篇(来自CSDN大佬)
import glob
import os
csv_list = glob.glob('*二手房信息.csv')
print(u'共发现%s个CSV文件' % len(csv_list))
print(u'正在处理............')
for i in csv_list:
fr = open(i, 'rb').read()
with open('result.csv', 'ab') as f:
f.write(fr)
print('合并完毕!')
运行结果
多写几次代码就会明白里面的逻辑,加油!