python - 如何在 pandas 数据框中指定列类型

Question

执行此行后

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

列“numbers”不是我期望的列表，而是它的类型对象，无法索引，并且我收到 IndexError。

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

如何修复它？

Answer 1

dtype of string、dict、list、set s，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'>

如果想检查长度s:

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