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

聪明小米

• 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("按下任意键关闭窗口...")
  

复制代码
很简单的读取表格提取数据的东西 大佬请手下留情

超级小爱

有点东西，蹲个实际体验