Python 我怎样才能得到这里所有的数据前缀

我想将所有数据的前两个字母存储在

prefixedSku

中，但只显示了3个结果

import pandas as pd
from pandas import Series, DataFrame

sdf = pd.read_csv('customers.csv')
skuList = sdf['sku'].values.tolist()
prefixedSku = {s[:2] for s in skuList}
print(prefixedSku)

我希望输出是这样的

{S2'，B1'，S1'，S2'，S1'，S1'，B1'，S2'，S1}

事实上，我有很多数据与S1、B1和S2有关
下面是customers.csv的类似示例

{'S2', 'B1', 'S1'}

---

如果我正确理解了这个问题，那么您需要一个与您的行对应的前缀列表。如果将集合理解切换为列表理解，则代码是正确的：

account  sku
0001     B1-20000
0002     S2-77896
        .
        .
        .
1000     S1-99999

---

编辑：将dict改为set

@dsbailey，它不是dict理解，而是set理解。集合只能具有唯一值，因此已删除重复项。

import pandas as pd
from pandas import Series, DataFrame

sdf = pd.read_csv('customers.csv') 
skuList = sdf['sku'].values.tolist()
# Changed {} to []
prefixedSku = [s[:2] for s in skuList]
print(prefixedSku)