猿问

Pandas 将字符串与 groupby 结合起来

我正在尝试在我的数据框中组合字符串。数据框如下所示:


0          code   text1

1        507489   text2

2        507489   text3

3        506141   text4

4        506141   text5

5        504273   text6

我当前的代码:


import pandas as pd


df = pd.read_csv("location.csv", header=None, delimiter=';', dtype='unicode', nrows=100)

new_header = df.iloc[0] 

df = df[1:] 

df.columns = new_header


df.groupby('code').agg('->'.join).reset_index()


df.to_csv (r'new_location\export_dataframe.csv', index = False, header=True)

print(df)

但我没有得到预期的结果。当我期待时,输出看起来与输入相同:


0          code   text1

1        507489   text2->text3

2        506141   text4->text5

3        504273   text6

对此很陌生,所以我一定犯了一些简单的错误。


产生相同结果的数据框:


testf = {'code': ['1','2','2','4'],

        'text': [22000,25000,27000,35000]

        }


df = pd.DataFrame(testf, columns = ['code', 'text'])


千巷猫影
浏览 171回答 1
1回答

杨__羊羊

似乎您忘记了分配回来,也被删除header=None了,read_csv因为在文件中是用于 DataFrame 中列名称的标题:import pandas as pddf = pd.read_csv("location.csv", sep=';', dtype='unicode', nrows=100)df = df.groupby('code').agg('->'.join).reset_index()print (df)     code         text10  504273         text61  506141  text4->text52  507489  text2->text3df.to_csv (r'new_location\export_dataframe.csv', index = False)
随时随地看视频慕课网APP

相关分类

Python
我要回答