Pandas数据匹配与列扩展：基于多列字符串的动态数据集成

1. 问题场景描述

在数据处理过程中，我们经常面临需要将分散在不同文件或数据表中的信息进行整合的挑战。一个典型的场景是： 我们拥有两个CSV文件：File_1.csv 和 File_2.csv。

• File_1.csv 包含多列结构信息，例如 Structure_1、Structure_2、Structure_3。这些结构值可能在 File_1 中重复出现。
• File_2.csv 包含一个唯一的 Structure 列及其对应的 Barcodes 列。

我们的目标是： 根据 File_1.csv 中 Structure_1、Structure_2、Structure_3 列的每个值，在 File_2.csv 的 Structure 列中查找匹配项。一旦找到匹配，就将 File_2.csv 中对应的 Barcodes 值作为新列 (Barcode_1、Barcode_2、Barcode_3) 添加到 File_1.csv 中。由于 File_1 中的结构值可能重复，因此新的条形码列也会相应地多次添加。

2. 解决方案概述：使用Pandas进行高效匹配

python的pandas库提供了强大的数据结构（dataframe）和数据分析工具，非常适合处理这类数据匹配和整合任务。本教程将重点介绍两种核心方法：

1. 构建映射字典 (Mapping Dictionary)：从源数据（File_2）创建一个键值对字典，其中键是用于匹配的结构字符串，值是对应的条形码。
2. 应用映射 (Applying Map)：利用DataFrame的 .map() 方法，将映射字典应用到目标数据（File_1）的相应列上，从而快速生成新的条形码列。

这种方法在处理大量数据时效率高，且代码简洁易懂。

3. 实现步骤与示例代码

3.1 准备工作：导入Pandas库与模拟数据

首先，我们需要导入Pandas库。为了方便演示，我们将创建两个模拟的DataFrame来代替实际的CSV文件。在实际应用中，您将使用 pd.read_csv() 来加载您的数据。

import pandas as pd

# 模拟 File_1.csv 的数据
# 假设 File_1 有 Structure_1, Structure_2, Structure_3 三列
df1 = pd.DataFrame({
    "Structure_1": ['CDAB', 'ACEG', 'KLMN', 'CDAB'],
    "Structure_2": ['ACEG', 'XAYBZ', 'ACEG', 'KLMN'],
    "Structure_3": ['KLMN', 'CDAB', 'XAYBZ', 'ACEG']
})

# 模拟 File_2.csv 的数据
# 假设 File_2 有 Structure 和 Barcodes 两列
df2 = pd.DataFrame({
    "Structure": ['CDAB', 'ACEG', 'KLMN', 'XAYBZ', 'PQRS'],
    "Barcodes": ['111', '222', '333', '444', '555']
})

print("原始 df1:")
print(df1)
print("\n原始 df2:")
print(df2)

输出示例：

原始 df1:
  Structure_1 Structure_2 Structure_3
0        CDAB        ACEG        KLMN
1        ACEG       XAYBZ        CDAB
2        KLMN        ACEG       XAYBZ
3        CDAB        KLMN        ACEG

原始 df2:
  Structure Barcodes
0      CDAB      111
1      ACEG      222
2      KLMN      333
3     XAYBZ      444
4      PQRS      555

3.2 构建映射字典

从 df2 中提取 Structure 和 Barcodes 列，构建一个字典。这个字典的键将是 Structure 值，值将是对应的 Barcodes 值。

# 从 df2 创建映射字典
# df2[['Structure', 'Barcodes']].values 将返回一个二维数组，
# dict() 函数可以直接将其转换为字典，其中第一列为键，第二列为值。
structure_to_barcode_map = dict(df2[['Structure', 'Barcodes']].values)

print("\n构建的映射字典:")
print(structure_to_barcode_map)

输出示例：

构建的映射字典:
{'CDAB': '111', 'ACEG': '222', 'KLMN': '333', 'XAYBZ': '444', 'PQRS': '555'}

3.3 应用映射并创建新列

现在，我们将这个映射字典应用到 df1 的 Structure_1、Structure_2 和 Structure_3 列上。Pandas DataFrame的 .map() 方法非常适合这种一对一（或多对一）的查找和替换操作。

Dreamhouse AI

AI室内设计，快速重新设计你的家，虚拟布置家具

78

查看详情

# 使用 .map() 方法为 df1 添加新的条形码列
df1['Barcode_1'] = df1['Structure_1'].map(structure_to_barcode_map)
df1['Barcode_2'] = df1['Structure_2'].map(structure_to_barcode_map)
df1['Barcode_3'] = df1['Structure_3'].map(structure_to_barcode_map)

print("\n处理后的 df1:")
print(df1)

输出示例：

处理后的 df1:
  Structure_1 Structure_2 Structure_3 Barcode_1 Barcode_2 Barcode_3
0        CDAB        ACEG        KLMN       111       222       333
1        ACEG       XAYBZ        CDAB       222       444       111
2        KLMN        ACEG       XAYBZ       333       222       444
3        CDAB        KLMN        ACEG       111       333       222

可以看到，df1 成功添加了 Barcode_1、Barcode_2 和 Barcode_3 三个新列，其值根据 Structure 列与 df2 中的映射关系填充。

4. 注意事项与最佳实践

• 处理未匹配值 (NaN)：如果 df1 中的某个结构值在 df2 的 Structure 列中没有对应的条形码，那么 .map() 方法将默认填充 NaN (Not a Number)。您可以根据需求对这些 NaN 值进行处理，例如使用 fillna() 填充默认值（如 0 或空字符串 ''），或者直接保留 NaN。

  # 示例：将未匹配的 NaN 填充为 'N/A'
  df1['Barcode_1'] = df1['Structure_1'].map(structure_to_barcode_map).fillna('N/A')

  登录后复制
• 性能考量：对于非常大的数据集，如果映射字典非常庞大，或者需要进行更复杂的匹配逻辑（例如多列联合匹配），除了 .map() 之外，Pandas的 merge() 函数也是一个强大的选择。merge() 适用于基于一个或多个键合并两个DataFrame，它在底层进行了优化，对于大数据量通常表现良好。然而，对于简单的单列查找映射，dict.map() 往往更简洁高效。
• 列名规范：在实际项目中，保持清晰一致的列命名规范非常重要。例如，新生成的条形码列命名为 Barcode_1, Barcode_2 等，明确指明其来源和关联性。
• 保存结果：处理完成后，通常需要将结果保存到新的CSV文件或其他格式。

  # 将结果保存到新的CSV文件
  df1.to_csv('File_3_output.csv', index=False)

  登录后复制

  index=False 参数用于避免将DataFrame的索引写入CSV文件。

5. 总结

本教程展示了如何利用Python Pandas库高效地解决跨文件数据匹配和列扩展的问题。通过构建映射字典和应用 .map() 方法，我们能够根据一个文件的字符串值，在另一个文件中查找并动态添加多个新列。这种方法不仅代码简洁，而且在处理大规模数据集时表现出良好的性能，是数据科学家和分析师在日常工作中常用的强大工具。掌握这种技术，将大大提升您的数据处理能力。

以上就是Pandas数据匹配与列扩展：基于多列字符串的动态数据集成的详细内容，更多请关注php中文网其它相关文章！