首頁 >後端開發 >Python教學 >按照列表中出現的唯一元素來標記列表

按照列表中出現的唯一元素來標記列表

PHPz
PHPz轉載
2024-02-08 20:57:04509瀏覽

按照列表中出現的唯一元素來標記列表

問題內容

給定字串列表,例如:

foo = \['a', 'a', 'b', 'a', 'b', 'c', 'c', 'a', 'b', 'c', 'a'\]

我們如何標記它們以使輸出為:

output = \['a1', 'a2', 'b1', 'a3', 'b2', 'c1', 'c2', 'a4', 'b2', 'c3', 'a5'\]

(保持原始列表順序)

在以下情況下,只有 3 個唯一變數需要查看,因此我首先想到的是查看唯一元素:

import numpy as np

np.unique(foo)

Output = \['A', 'B', 'C'\]

但是當我試圖找到合適的循環來達到所需的輸出時,我就會陷入困境。


正確答案


使用純python,利用字典來計算值:

foo = ['a', 'a', 'b', 'a', 'b', 'c', 'c', 'a', 'b', 'c', 'a']

d = {}
out = []
for val in foo:
    d[val] = d.get(val, 0)+1
    out.append(f'{val}{d[val]}')

如果您可以使用

import pandas as pd

s = pd.Series(foo)
out = s.add(s.groupby(s).cumcount().add(1).astype(str)).tolist()

輸出:['a1', 'a2', 'b1', 'a3', 'b2', 'c1', 'c2', 'a4', 'b3', 'c3', ' a5' ]

以上是按照列表中出現的唯一元素來標記列表的詳細內容。更多資訊請關注PHP中文網其他相關文章!

陳述:
本文轉載於:stackoverflow.com。如有侵權,請聯絡admin@php.cn刪除