读取csv文件提取指定列的数据

查看 44|回复 3
作者:jia222   
[Python] 纯文本查看 复制代码import glob
import os
import pandas as pd
# 获取当前目录下所有CSV文件路径
csv_files = glob.glob("*.csv")
# 遍历每个CSV文件
for csv_file in csv_files:
    # 获取CSV文件名(不包含扩展名)
    file_name = os.path.splitext(csv_file)[0]
   
    # 读取CSV文件并提取第1列和第2列的数据
    try:
        data = pd.read_csv(csv_file, encoding='utf-8')
    except UnicodeDecodeError:
        try:
            data = pd.read_csv(csv_file, encoding='gbk')
        except:
            print(f"无法读取CSV文件:{csv_file},请检查编码方式。")
            continue
   
    m_column = data.iloc[:, 1]  # 第2列的索引为1
    am_column = data.iloc[:, 0]  # 第1列的索引为0
    # 创建一个新的DataFrame来保存提取的数据
    new_data = pd.DataFrame({"手机号": am_column, "姓名": m_column})
   
    # 去除重复数据,保持每行对应关系
    new_data = new_data.drop_duplicates()
   
    # 删除手机号为空的行
    new_data = new_data.dropna(subset=["手机号"])
    # 将数据保存到新的xlsx文件中,文件名与所读取的CSV文件名一致
    output_file = f"{file_name}.xlsx"
    new_data.to_excel(output_file, index=False)
    print(f"已保存文件:{output_file}")
# 等待用户输入任意键后关闭窗口
input("按下任意键关闭窗口...")
很简单的东西,大佬请略过.

文件, 数据

wangguang   

数据分析学的很厉害呀
xuanwu126   

学习一下,以后应该会用到
wapjsx   

求问,为什么不直接 用 csv 库呢?
您需要登录后才可以回帖 登录 | 立即注册

返回顶部