Python：根据列名和条件替换数据框值

Question

假设我们有一个包含6列的数据框。 对于这些列中3个的子集，我们希望将实例3更改为1。您能建议一种优雅的方法吗？ 只需像这样写一行：

df['A'][df.A == 3] = 1 

由于维度大得多（例如在200个变量中更改50个），因此在我的实际数据中效率较低。

也许我们可以研究这个例子：

df = pd.DataFrame(np.random.randint(0,10,size=(100, 6)), columns=list('ABCDEF'))

我们要在A，B和C列中将值更改为3到1的地方。谢谢。

Answer 1

您可以遍历列的子集，并使用loc进行分配。

subset cols = ['A', 'B', 'C']  # For example.
for col in subset_cols:
    df.loc[df[col] == 3, col] = 1