从现有的两列python中创建唯一ID

Question

我的问题是：如何从现有的id列有效地签署数据唯一ID号？ 例如：我有两列[household_id]和[person_no]。 我尝试创建一个新列，查询将是：household_id +'_'+ person_no。

这是一个示例：

hh_id       pno  
 682138    1   
 365348    1     
 365348    2

想拿到：

unique_id
682138_1
365348_1
365348_2

并将此unique_id添加为新列。 我正在应用Python。 我的数据非常大。 任何有效的方法都会很棒。 谢谢！

Answer 1

你可以使用熊猫 。

假设您的数据位于csv文件中，请读入数据：

import pandas as pd 

df = pd.read_csv('data.csv', delim_whitespace=True)

创建新的id列：

df['unique_id'] = df.hh_id.astype(str) + '_' + df.pno.astype(str)

现在df看起来像这样：

    hh_id  pno unique_id
0  682138    1  682138_1
1  365348    1  365348_1
2  365348    2  365348_2

写回csv文件：

df.to_csv('out.csv', index=False)

文件内容如下所示：

hh_id,pno,unique_id
682138,1,682138_1
365348,1,365348_1
365348,2,365348_2