Python拆分csv文件

运用Python对csv文件的某一列中的不同内容进行拆分后自动生成不同的表格（Python运行代码、csv文件在同一文件夹下，生成的csv文件也在该文件夹里）

Python运行代码如下：

# -*- coding: utf-8 -*-
"""
Created on Tue Jul  2 15:44:43 2024

@author: wlpc
"""


import pandas as pd  
import os  
  
# 读取CSV文件  
file = '2024-06-26 17 57 43_hoho_18~22_2_raw_50.csv'
df = pd.read_csv(file,index_col=False,error_bad_lines=False)  
  
# 假设你想要拆分的列名为'IntegerColumn'，并且它包含的是整数的字符串形式  
# 注意：如果列名不是'IntegerColumn'，请替换为你实际的列名  
column_to_split = 'num'  
  
# 将字符串转换为整数，并获取唯一的整数值  
unique_integers = df[column_to_split].astype(int).unique()  
  

# 遍历唯一的整数值，并为每个值创建一个新的DataFrame  
for integer in unique_integers:  
    # 使用布尔索引来选择与当前整数匹配的行  
    df_subset = df[df[column_to_split].astype(int) == integer]  
      
    # 构造输出文件的名称，例如 output_1.csv, output_2.csv, ...  
    output_filename = f'{file[0:20]}hoho_{integer}_2_raw_10.csv'  
      
    # # 确保输出文件路径存在  
    # output_dir = os.path.dirname(output_filename)  
    # if not os.path.exists(output_dir):  
    #     os.makedirs(output_dir)  
      
    # 将拆分后的DataFrame保存到CSV文件中  
    df_subset.to_csv(output_filename, index=False)  
    print(f'Saved {output_filename}')

对num列不同的数据进行拆分

拆分前

拆分后