Python 3.x 保留数据框中所有列表的方括号
我有一个函数,可以在包含二维列表的列表中查找元素的可能组合。然后,所有组合在数据帧中表示Python 3.x 保留数据框中所有列表的方括号,python-3.x,pandas,dataframe,Python 3.x,Pandas,Dataframe,我有一个函数,可以在包含二维列表的列表中查找元素的可能组合。然后,所有组合在数据帧中表示x1_df #Function to generate combinations def Generate1(rv): rv = [rv] x1 = [np.array(np.meshgrid(*rv[i]), dtype = object).T.reshape(-1,len(rv[i])) for i in range(len(rv))] x1_df = pd.DataFrame(x
x1_df
#Function to generate combinations
def Generate1(rv):
rv = [rv]
x1 = [np.array(np.meshgrid(*rv[i]), dtype = object).T.reshape(-1,len(rv[i])) for i in range(len(rv))]
x1_df = pd.DataFrame(x1[0], index=range(x1[0].shape[0]),columns=range(x1[0].shape[1]))
return x1_df
列表(函数中的参数)。此列表中的所有元素都有一个括号,但它会删除dataframe中带有单个元素的括号
rv = [[[-5], [0], [-5, 0]], [[0]], [[-999]]]
输出:
包含单个元素的列表中缺少括号
预期输出:对于具有单个元素的列表,如何在dataframe中保留方括号
在return语句中尝试
返回x1_df.applymap(lambda x:x如果是instance(x,list)else[x])
?
print(Generate1(rv)) #Call function
0 1 2
0 [-5] 0 -999
1 [0] 0 -999
2 [-5, 0] 0 -999
0 1 2
0 [-5] [0] [-999]
1 [0] [0] [-999]
2 [-5, 0] [0] [-999]