python删除数据集含有a的列
假设我们有一个数据集,数据集包含多个列,我们需要删除包含字母 "a" 的列,可以使用以下代码:
data = {"col1": [1, 2, 3], "col2": ["a", "b", "c"], "col3": [4, 5, 6], "col4": ["d", "e", "f"]}
df = pd.DataFrame(data)
cols_to_drop = []
for col in df.columns:
if "a" in col:
cols_to_drop.append(col)
df.drop(cols_to_drop, axis=1, inplace=True)
在上面的代码中,我们首先创建了一个包含多个列的数据集。然后,我们迭代所有的列,并检查每个列是否包含 "a"。如果包含,则将该列添加到一个列表中。最后,我们使用 drop 函数删除包含 "a" 的列。axis=1 指定我们要删除列而不是行,inplace=True 指定我们要在原始数据帧中进行修改。
注意,为了运行上面的代码,您需要先导入 Pandas 库
原文地址: https://www.cveoy.top/t/topic/fyRS 著作权归作者所有。请勿转载和采集!