Question

我想确定pandas中的列是否是列表（在每一行中）。

df=pd.DataFrame({'X': [1, 2, 3], 'Y': [[34],[37,45],[48,50,57]],'Z':['A','B','C']})

df
Out[160]: 
   X             Y  Z
0  1          [34]  A
1  2      [37, 45]  B
2  3  [48, 50, 57]  C

df.dtypes
Out[161]: 
X     int64
Y    object
Z    object
dtype: object

由于字符串的dtype是“object”，我无法区分字符串和列表（整数或字符串）。

如何识别列“Y”是否为int？

列表

Answer 1

您可以使用applymap，进行比较，然后添加all以检查所有值是否为True s：

print (df.applymap(type))
               X               Y              Z
0  <class 'int'>  <class 'list'>  <class 'str'>
1  <class 'int'>  <class 'list'>  <class 'str'>
2  <class 'int'>  <class 'list'>  <class 'str'>

a = (df.applymap(type) == list).all()
print (a)
X    False
Y     True
Z    False
dtype: bool

或者：

a = df.applymap(lambda x: isinstance(x, list)).all()
print (a)
X    False
Y     True
Z    False
dtype: bool

如果需要列表列表：

L = a.index[a].tolist()
print (L)
['Y']

如果要检查dtypes（但strings，list，dict为object s）：

print (df.dtypes)
X     int64
Y    object
Z    object
dtype: object

a = df.dtypes == 'int64'
print (a)
X     True
Y    False
Z    False
dtype: bool

Answer 2

如果数据集很大，则应在应用类型函数之前先进行抽样，然后可以检查：

如果最常见的类型是列表：

df\
.sample(100)\
.applymap(type)\
.mode(0)\
.astype(str) == "<class 'list'>"

如果所有值都是列表：

(df\
.sample(100)\
.applymap(type)\
.astype(str) == "<class 'list'>")\
.all(0)

如果列表中有任何值：

(df\
.sample(100)\
.applymap(type)\
.astype(str) == "<class 'list'>")\
.any(0)

如何识别pandas列是一个列表

2 个答案: