python - 查找 Pandas DataFrame 中包含字典的列

Question

我有一个 pandas DataFrame，其中有几个包含 dicts 的列。我正在尝试识别至少包含 1 个字典的列。

import pandas as pd
import numpy as np 

df = pd.DataFrame({
                   'i': [0, 1, 2, 3],
                   'd': [np.nan, {'p':1}, {'q':2}, np.nan],
                   't': [np.nan, {'u':1}, {'v':2}, np.nan]
                 })


 # Iterate over cols to find dicts
 cdict = [i for i in df.columns if isinstance(df[i][0],dict)]
 cdict
 [] 
 
 

如何使用 dicts 查找列？是否有解决方案可以使用 dicts 查找列，而无需迭代每个单元格/列的值？

Answer 1

你可以这样做:

s = df.applymap(lambda x:isinstance(x, dict)).any()
dict_cols = s[s].index.tolist()

打印(dict_cols)

['d', 't']