如何使用Pandas提取两个文件的指定列并生成data.csv

实现步骤 #

• 导入Pandas库
• 构造示例中的两个数据集
• 分别提取所需列
• 合并列数据
• 输出为CSV文件

完整代码 #

import pandas as pd

# 构造示例1的数据集
df1 = pd.DataFrame({
    'district_id': [18, 1],
    'date': ['1995-03-24', '1993-02-26']
})

# 构造示例2的数据集
df2 = pd.DataFrame({
    'link_id': [9, 19],
    'type': ['gold', 'classic']
})

# 提取指定列并按列合并
merged_df = pd.concat([df1['date'], df2['type']], axis=1)

# 输出到data.csv，不写入索引列
merged_df.to_csv('data.csv', index=False)

补充说明 #

• 当两个数据集行数一致时，pd.concat配合axis=1可以直接按行索引对齐合并列
• index=False参数用于避免将Pandas的默认索引写入最终的CSV文件

内容的提问来源于stack exchange，提问作者QZhu