在熊貓中將字符串列表轉換為int列表

Question

讀取CSV文件后，我有一列包含一串用逗號分隔的十六進制值。 例如，“ 0x00000034,0x0000008a，0x00000145”。 我想將此列轉換為整數列表。 我可以使用以下代碼拆分並轉換為整數列表：

for index, row in df.iterrows():
    rnti_list = [int(idx, 16) for idx in row['rnti'].split(",")]

然后，我可以將其添加為新列。

想知道最優雅的代碼，用最少的行數就能完成相同的工作。

謝謝。

Answer 1

使用嵌套列表理解：

df = pd.DataFrame({'rnti':['0x00000034,0x0000008a,0x00000145','0x00000034,0x0000008a']})
print (df)
                               rnti
0  0x00000034,0x0000008a,0x00000145
1             0x00000034,0x0000008a

rnti_list = [[int(idx, 16) for idx in x.split(",")] for x in df['rnti']]
print (rnti_list)
[[52, 138, 325], [52, 138]]

---

df['new'] = [[int(idx, 16) for idx in x.split(",")] for x in df['rnti']]
print (df)
                               rnti             new
0  0x00000034,0x0000008a,0x00000145  [52, 138, 325]
1             0x00000034,0x0000008a       [52, 138]

Answer 2

使用apply

例如：

import pandas as pd

df = pd.DataFrame({"rnti": ["0x00000034,0x0000008a,0x00000145"]})    
print( df["rnti"].str.split(",").apply(lambda x: [int(i, 16) for i in x]) )

輸出：

0    [52, 138, 325]
Name: rnti, dtype: object