Python提取行首数字的内容到新文件夹

在Python中，你可以使用pandas库来读取文件并提取行首的数字作为索引，然后将结果写入一个新的文件。这里是一个简单的步骤：

import pandas as pd

# 假设你的数据存储在一个CSV文件中
file_name = 'your_file.csv'

# 使用pandas读取文件
df = pd.read_csv(file_name)

# 提取行首的数字（假设它们都在第一列）
if df.columns[0].startswith('数字'):
    # 如果列名以'数字'开头，假设它是数字列
    index_numbers = df.iloc[:, 0].str.extract(r'\d+', expand=False)
else:
    # 如果不是，检查是否有连续的数字
    index_numbers = df.apply(lambda x: int(x[x.str.isdigit()]), axis=1)

# 创建新的目录，如果不存在
new_dir = 'new_folder'
if not os.path.exists(new_dir):
    os.makedirs(new_dir)

# 将索引应用到原始数据上，并保存到新文件夹
new_filename = os.path.join(new_dir, 'extracted_data.csv')
df.index = index_numbers
df.to_csv(new_filename, index=True)