从现有的两列 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