我需要幫助使用 Python 從 csv 文件中識別數據子集。我有一個包含水果清單和類型字段的數據集，如下所示。Item Type Apple 10 Orange 5 Pear 8 Apple 10 Pear 5Orange 2Pear 5我需要能夠將此數據集識別/拆分為水果和類型的子集。預期輸出是另一個帶有數字的字段，用于指示它應該進入的類別： Item Type Category Apple 10 1 Orange 5 2 Pear 8 3 Apple 10 1 Pear 5 4 Orange 5 2 Pear 5 4我是 Python 新手，因此需要有關實際邏輯的幫助
2 回答
慕森王
TA貢獻1777條經驗 獲得超3個贊
我將假設預期答案的倒數第二行應保留為“橙色 2”，然后鍵入：
import pandas
df = pandas.DataFrame([['Apple', '10'],
['Orange', '5'],
['Pear', '8'],
['Apple', '10'],
['Pear', '5'],
['Orange', '2'],
['Pear', '5']],
columns = ['Item', 'Type'])
categories = []
for index,row in df.iterrows():
if tuple(row) not in categories:
categories.append(tuple(row))
df["Category"] = [categories.index(tuple(row))+1 for index,row in df.iterrows()]
print(df)
這會給
Item Type Category
0 Apple 10 1
1 Orange 5 2
2 Pear 8 3
3 Apple 10 1
4 Pear 5 4
5 Orange 2 5
6 Pear 5 4
由于這里沒有特殊的 Pandas 操作，因此沒有必要使用 Pandas（盡管它可能對其他操作有用）。關鍵是定義一個唯一的元組列表，記錄Item和Type的每個組合，并使用index函數再次查找。
隔江千里
TA貢獻1906條經驗 獲得超10個贊
我真的不明白你的問題。
如果要讀取 csv 文件，可以使用pandas.
import pandas as pd
pd.read_csv('myfile.csv')
您還可以使用sep參數來使用自定義分隔符。
添加回答
舉報
0/150
提交
取消