Python Day8作业,批量导入CSV文件数据

批量导入CSV文件数据(初级篇)

先导入需要读取得文件夹‘成都链家二手房数据’
Python Day8作业,批量导入CSV文件数据_第1张图片

初级篇

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('写入完成')

运行结果
Python Day8作业,批量导入CSV文件数据_第2张图片

高级篇(来自CSDN大佬)

import glob
import os

csv_list = glob.glob('*二手房信息.csv')  # 查看同文件夹下的csv文件数
print(u'共发现%s个CSV文件' % len(csv_list))
print(u'正在处理............')
for i in csv_list:  # 循环读取同文件夹下的csv文件
    fr = open(i, 'rb').read()
    with open('result.csv', 'ab') as f:  # 将结果保存为result.csv
        f.write(fr)
print('合并完毕!')

运行结果
Python Day8作业,批量导入CSV文件数据_第3张图片
多写几次代码就会明白里面的逻辑,加油!

你可能感兴趣的:(python练习,初级Python,python,大数据,开发语言)