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

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 库呢？

zhengsg5

pandas感觉确实好用，数据处理非常方便！

HR741158

学习了，收藏备用

greatzdl

感觉这小代码还是比较实用哦

MAOSKE

感谢分享！