根據條件拆分列表（性別）

Question

我是一名Python初學者，剛剛學習如何編寫函數。我有一份權重和性別列表，並試圖根據性別標準將其分成兩個新列表。使用for循環，我已經成功至今：

df = pd.read_csv('brainweight.csv') 
w = list(df['Weight']) 
s = list(df['Sex']) 

female_weight = [] 
male_weight = [] 

for sex, weight in zip (s, w): 
    if sex == 'f': 
     female_weight.append(weight) 
    else: 
     male_weight.append(weight) 

我應該如何修改這個for循環到一個函數，其中變量= M/F（性別）？

Answer 1

df = pd.read_csv('brainweight.csv') 
w = list(df['Weight']) 
s = list(df['Sex']) 
def list_by_gender(): 
    females = [] 
    males = [] 
    for sex, weight in zip (s, w): 
     if sex == 'f': 
      females.append(weight) 
     else: 
      males.append(weight) 
    return males,females 

male_list,female_list=list_by_gender()