python - 使用 pandas 根据与另一个单元格的比较来更改每个单元格中的值

Question

我想将第 0 列中的两个值与所有其他列中的值进行比较，并相应地更改为这些列的值。我有 4329 行 x 197 列。

从此:

  0  1  2  3
0 G  G  G  T
1 A  A  G  A
2 C  C  C  C
3 T  A  T  G

对此:

  0  1  2  3
0 G  1  1  0
1 A  1  0  1
2 C  1  1  1
3 T  0  1  0

我尝试过嵌套 for 循环，但它不起作用并且速度很慢。

for index, row in df.iterrows():        
    for name, value in row.iteritems():
        if name == 0:
            c = value
            continue
        if value == c:
            value = 1
        else:
            value = 0

我无法拼凑出使用 apply 或 applymap 来解决问题的方法。

Answer 1

这是使用 iloc 和 eq 的方法:

df.iloc[:,1:] = df.iloc[:,1:].eq(df.iloc[:,0], axis=0).astype(int)

输出:

   0  1  2  3
0  G  1  1  0
1  A  1  0  1
2  C  1  1  1
3  T  0  1  0