创建一个列表列表，其中列表的长度在另一列中提到

Question

我的df看起来像这样;

data = pd.DataFrame({'count':[1,2,4,5,8,1,3]})

如何创建一个名为'lists'的新列，其中包含一个随机列表，其中列表的长度是'count'列中的值？像这样......

data = pd.DataFrame({'count':[1,2,4,5,8,1,3],
                     'lists': [[1],[34,56],[25,65,1,4],[33,37,89,12,11],[45,56,78,98,12,2,3,4],[234],[23,456,765]]

Answer 1

你可以尝试：

data['list'] = data.applymap(lambda x: random.sample(xrange(100), x))

可替换地，

data['list'] = data['count'].apply(lambda x: pd.np.random.random_integers(1, 100, x) )