如何在熊猫数据框中指定列类型

Question

执行此行后

data['numbers'] = data.apply(lambda row : [1] * len(row.text), axis=1)

“数字”列不是我期望的列表，而是无法索引的对象类型，我得到 IndexError。

我想要的结果是一个带有“数字”的列，其中每一行的数字与该行中相应文本的长度一样多。

我该如何解决？

Answer 1

dtype of strings、dicts、lists、sets、tuples 始终为object，用于测试type：

data = pd.DataFrame({'text':['aaas','as']}, index=[10,12])

data['numbers'] = data.apply(lambda row : [1] * len(row.text), axis=1)

print (data['numbers'].apply(type))
0    <class 'list'>
1    <class 'list'>
Name: numbers, dtype: object

#check scalar
print (type(data.loc[0, 'numbers']))
<class 'list'>

如果要查看lengths:

print (len(data.iloc[0, data.columns.get_loc('numbers')]))
4

data['lens'] = data['numbers'].str.len()
print (data)

    text       numbers  lens
10  aaas  [1, 1, 1, 1]     4
12    as        [1, 1]     2