將系列列表連接到一個uid

Question

我有一個Pandas數據框，其中有幾列組成一個唯一標識符。我想寫一個通用測試用例，允許我將這些列連接在一起成爲一個列（uid）並測試該列的唯一性。我有以下代碼作爲非通用測試用例：

import pandas as pd 
import pytest 
df = pd.DataFrame(columns=['one', 'two', 'three']) 
df.one = 'abc', 'def', 'ghi' 
df.two = 'jkl', 'mno', 'pqr' 
df.three = 'stu', 'vwx', 'yzz' 
# Test one 
df['uid'] = df.one + df.two 
assert len(df.index) == len(df.drop_duplicates(['uid']).index) 
# Test two 
df['uid'] = df.one + df.three 
assert len(df.index) == len(df.drop_duplicates(['uid']).index) 

由於我將重用這一點，我需要一種解決方案，其允許選擇任意的列的要連接在一起，例如在管線8和11.

Answer 1

假設你要選擇列two和three補充：

col_to_add = ['two', 'three'] 

使用sum(axis=1)來連接這些列：

df['uid'] = df[col_to_add].sum(axis=1)