基於行條件在 Pandas 數據框中刪除列

Question

我有以下熊貓數據框，我想刪除“貿易”行中任何具有 0-Flat 的列。 我該怎么做呢？

Answer 1

這應該做的工作

headers_to_drop=[header for header in df.columns if (df[header]=="0-Flat").any()]
df=df.drop(headers_to_drop,axis=1)

代碼首先遍歷所有列並檢查是否有包含字符串的列並刪除選定的列。

Answer 2

您可以在dropna()旁邊使用replace來查看整個 df 並刪除包含 0-Flat 字符串的所有列

import numpy as np
import pandas as pd

df = pd.DataFrame({
    'Column1' : ['1', '2', '3', '0-Flat'],
    'Column2' : ['1', '2', '3', '4'],
    'Column3' : ['1', '2', '0-Flat', '0']
})

df.replace('0-Flat', np.nan).dropna(axis = 1)