python - 如何编辑数据框的输出？

Question

import csv
import pandas as pd
data = {'numbers' : [1, 2, 3, 1, 8, 4, 5, 7, 3, 6, 2, 7, 4, 5, 8]
'colors' : ['red', 'yellow', 'orange', 'green', 'blue', 'purple', 'magenta', 'grey', 'pink', 'cyan', 'lime', 'apricot', 'teal', 'navy', 'maroon']}

df = pd.DataFrame(data, columns = ['numbers', 'colors'])    
temp = df.groupby(['numbers'])
temp1 = temp.sum()['colors']
print(temp1)

目前，我的输出是这样的:

1 redgreen
2 yellowlime
3 orangepink

我希望能够用箭头格式化我的输出:

1 red --> green
2 yellow --> lime
3 orange --> pink

它还应该能够处理 3 个或更多值，例如:

10 value1 --> value2 --> value3 --> value4

如果有任何帮助，我将不胜感激，谢谢!

Answer 1

在一行中，您可以:

df.groupby('numbers').agg({'colors': lambda x: ' --> '.join(x)})

如果你打印这个，你会得到:

                   colors
numbers                  
1           red --> green
2         yellow --> lime
3         orange --> pink
4         purple --> teal
5        magenta --> navy
6                    cyan
7        grey --> apricot
8         blue --> maroon

更简单，感谢@piRSquared

df.groupby('numbers').colors.apply(' --> '.join)