Python 将元素插入 Pandas Dataframe 列表

发布于05月23日

具有以下DataFrame:

df = pd.DataFrame({"booking": ['A', 'B', 'C'],
                   "route": [['1','2','3'],['5','4','1'],['3','6','8']]})

df1 = pd.DataFrame({"booking": ['A', 'B', 'C'],
                    "route": [['0','1','2','3'],['0','5','4','1'],['0','3','6','8']]})

df1 = pd.DataFrame({"booking": ['A', 'B', 'C'],
                    "legs": [['0>1','1>2','2>3'],['0>5', '5>4','4>1'],['0>3','3>6','6>8']]})

推荐答案

您可以使用列表理解:

df1 = df.assign(route=[[0]+l for l in df['route']])

输出:

  booking         route
0       A  [0, 1, 2, 3]
1       B  [0, 5, 4, 1]
2       C  [0, 3, 6, 8]

然后:

df1 = df.assign(route=[[f'{a}>{b}' for a,b in zip(l, l[1:])]
                       for l in df1['route']])

输出:

  booking            route
0       A  [0>1, 1>2, 2>3]
1       B  [0>5, 5>4, 4>1]
2       C  [0>3, 3>6, 6>8]

当然，正如@nick所 comments 的那样，如果不需要中间件，您可以使用单个步骤:

df1 = df.assign(route=[[f'{a}>{b}' for a,b in zip([0] + l, l)]
                       for l in df['route']])

输出:

  booking            route
0       A  [0>1, 1>2, 2>3]
1       B  [0>5, 5>4, 4>1]
2       C  [0>3, 3>6, 6>8]

实用课程推荐