如何创建两列值相互关联的DataFrame？

核心思路 #

针对你提出的"某列出现数字2时，相邻对应行设为0"的需求，用Pandas可以通过两种方式实现，以下是具体操作：

方法1：布尔索引直接赋值（高效推荐） #

适合简单依赖逻辑，处理速度快，大数据量场景优先用这种方式。

步骤1：创建示例DataFrame #

import pandas as pd

df = pd.DataFrame({
    'col_a': [1, 2, 3, 2, 5],
    'col_b': [4, 5, 2, 6, 7]
})

步骤2：设置依赖规则 #

# 当col_a等于2时，将对应行的col_b设为0
df.loc[df['col_a'] == 2, 'col_b'] = 0

# 当col_b等于2时，将对应行的col_a设为0
df.loc[df['col_b'] == 2, 'col_a'] = 0

处理后结果 #

方法2：逐行处理（适合复杂逻辑） #

如果依赖规则更复杂（比如涉及多列联动、自定义判断条件），可以用apply逐行处理：

def adjust_dependent_cols(row):
    # 自定义依赖逻辑：col_a为2则col_b设0，col_b为2则col_a设0
    if row['col_a'] == 2:
        row['col_b'] = 0
    if row['col_b'] == 2:
        row['col_a'] = 0
    return row

# 应用规则到每一行
df = df.apply(adjust_dependent_cols, axis=1)

扩展场景说明 #

如果是多列相邻依赖（比如col2为2时，同时将col1和col3设为0），可以直接扩展布尔索引的列选择：

# 当col2等于2时，将col1和col3对应行设为0
df.loc[df['col2'] == 2, ['col1', 'col3']] = 0

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