pandas使用iloc获取索引列表的方法?

时间:2018-05-21 20:21:46

标签: python pandas vectorization

我按照我想要的方式对数据进行了排序。 我即将投入类似的内容:

class

所以该列将是:

series_data = []
for count,x in enumerate(df):
   series_data.append(list(range(count)))
   df['up_to_row'].iloc(count)= series_data

然后我需要将这个iloc位置转换为索引。

是否有更多熊猫特定的方法来做到这一点? 我会用pandas操作员做这个,但我不知道如何获得当前的iloc(这就是为什么需要枚举)

编辑* 使用@wen的一些工具找到了最终答案。谢谢。

df['up_to_row'] = Series([0], [0,1], [0,1,2], [0,1,2,3]...)

1 个答案:

答案 0 :(得分:1)

使用cumsum,注意它会将列表中的数字转换为str,而不是int。

df['up_to_row']=np.arange(len(df))
(df['up_to_row'].astype(str)+',').cumsum().str[:-1].str.split(',')
Out[211]: 
0             [0]
1          [0, 1]
2       [0, 1, 2]
3    [0, 1, 2, 3]
Name: up_to_row, dtype: object