python - Python - 如果列名包含特定字符串，则更改该列中的值，否则值保持不变

Question

我正在尝试编写一些代码来查看列名并将列值设置为 0 如果该列名包含特定字符串。我有一个与此类似的列的数据集：

date apples_sold bananas_sold pineapples_sold orange_sold

我想用字符串“apple”更新任何列，以使该列的每一行都为 0。如果该列不包含“apple”，则该值应保留。

我尝试遍历列名，但不确定语法，因为我不想硬编码特定的列名。

标签： pythonpandasdataframe

Answer 1

尝试loc更新.str.contains：

df.loc[:, df.columns.str.contains('apple')] = 0