python - 查找至少一行包含字母的列

Question

假设我有以下数据集:

import pandas as pd

df = pd.DataFrame(
        {'A': [1, 2, 3],
         'B': ['one', 2, 3],
         'C': [4, 5, '6Y']
         })

我想找出 - 无需任何繁琐的 for 循环 - 哪些列至少包含一个带字母的大小写(此处:B 和 C)。我想结果应该是 bool 值列表或索引。

感谢您的帮助!

Answer 1

作为一种快速简单的解决方案，您可以使用replace 和 filter:

df.replace('(?i)[a-z]', '', regex=True).ne(df).any()

A    False
B     True
C     True
dtype: bool

df.columns[df.replace('(?i)[a-z]', '', regex=True).ne(df).any()]
# Index(['B', 'C'], dtype='object')

---

另一种选择是按列应用 str.contains:

mask = df.astype(str).apply(
    lambda x: x.str.contains(r'[a-z]', flags=re.IGNORECASE)).any()
mask

A    False
B     True
C     True
dtype: bool

df.columns[mask]
# Index(['B', 'C'], dtype='object')