объединить строки из CSV с питоном

Question

Например, приведенный ниже фрейм данных у меня есть

jan,feb,mar
AAA,BBB,CCC
Dog,Cat,Ant
111,222,333
Mon,Tue,Wed
1990,1991,1991
XXX,YYY,ZZZ

Я хотел бы добавить строку 1 и строку 2 к первой строке и т. Д.

jan,feb,mar,AAA,BBB,CCC,Dog,Cat,Ant
AAA,BBB,CCC,Dog,Cat,Ant,111,222,333
Dog,Cat,Ant,111,222,333,Mon,Tue,Wed
111,222,333,Mon,Tue,Wed,1990,1991,1991
Mon,Tue,Wed,1990,1991,1991,XXX,YYY,ZZZ
1990,1991,1991
XXX,YYY,ZZZ

, и если у меня естьтысяча строк, как я могу получить ответ

спасибо за ваш ответ

Answer 1

Используйте pd.concat с axis=1 и DataFrame.shift().

n = 3
df = pd.concat((df.shift(-i) for i in range(n)), 1)

Полный пример:

import pandas as pd

data = '''\
jan,feb,mar
AAA,BBB,CCC
Dog,Cat,Ant
111,222,333
Mon,Tue,Wed
1990,1991,1991
XXX,YYY,ZZZ'''

fileobj = pd.compat.StringIO(data) # replace this with 'path/to/file'
df = pd.read_csv(fileobj, sep=',', header=None)

n = 3
df = pd.concat((df.shift(-i) for i in range(n)), 1)

# And some formatting...
df.columns = range(len(df.columns))
df.fillna('', inplace=True)

print(df)
# df.to_csv('output.csv', index=False, header=None) # output if you want.

Возвращает:

      0     1     2     3     4     5     6     7     8
0   jan   feb   mar   AAA   BBB   CCC   Dog   Cat   Ant
1   AAA   BBB   CCC   Dog   Cat   Ant   111   222   333
2   Dog   Cat   Ant   111   222   333   Mon   Tue   Wed
3   111   222   333   Mon   Tue   Wed  1990  1991  1991
4   Mon   Tue   Wed  1990  1991  1991   XXX   YYY   ZZZ
5  1990  1991  1991   XXX   YYY   ZZZ                  
6   XXX   YYY   ZZZ