Python pandas 连接列 csv

Question

我有一个巨大的 Users_id 列表，我想要连接它们。我知道如何在 Excel 中执行此操作，但文件太大。

Users ID    
101 101
102 101,102
103 101,102,103
104 101,102,103,104

这就是我想要实现的目标。这是我到目前为止所拥有的。

import pandas as pd

df = pd.read_csv('file.csv')

pd.concat = df['USER ID']=.astype(str)+','+df['USER ID']

Answer 1

这是一个不寻常的操作，因为您的输入是数字，而输出是一系列以逗号分隔的字符串。一种解决方案是使用 itertools.accumulate使用 f 字符串(Python 3.6；PEP498):

import pandas as pd
from itertools import accumulate

df = pd.DataFrame({'Users': [101, 102, 103, 104]})

def joiner(x, y):
    return f'{x},{y}'

df['Cumulative'] = list(accumulate(df['Users'].astype(str), func=joiner))

print(df)

   Users       Cumulative
0    101              101
1    102          101,102
2    103      101,102,103
3    104  101,102,103,104