我有一個包含列（名稱，a，b）的數據框，我想創建一個列名稱“mean”，這將是列 a 和 b 的平均值，但如果任何兩行的平均值相同，則其總和值為更多應該減少0.1。 data frame 1 Name Sum a b mean0 hamm 34 2 2 21 jam 54 1 1 --> 12 tan 36 3 1 23 pan 39 4 4 4正如我們現在看到的那樣，第 0 行和第 2 行具有相同的平均值，所以現在誰的總和值更大應該減少 0.1在這里，在這種情況下，它的第 2 行的值應該是 2- 0.1 = 1.9最后結果 Name Sum a b mean0 hamm 34 2 2 21 jam 54 1 1 12 tan 36 3 1 1.93 pan 39 4 4 4
1 回答
牛魔王的故事
TA貢獻1830條經驗 獲得超3個贊
df.assign(mean = df[["a", "b"]].mean(axis=1))\
.assign(mean = df["mean"].subtract(df.groupby("mean").cumcount().divide(10)))
輸出
# Name Sum a b mean
# 0 hamm 34 2 2 2.0
# 1 jam 54 1 1 1.0
# 2 tan 36 3 1 1.9
# 3 pan 39 4 4 4.0
說明：
完整代碼+插圖
# Step 1
df["mean"] = df[["a", "b"]].mean(axis=1)
print(df)
# Name Sum a b mean
# 0 hamm 34 2 2 2.0
# 1 jam 54 1 1 1.0
# 2 tan 36 3 1 2.0
# 3 pan 39 4 4 4.0
# Step 2.1 + 2.2
print(df.groupby("mean").cumcount())
# 0 0
# 1 0
# 2 1
# 3 0
# dtype: int64
# Step 2.3
print(df.groupby("mean").cumcount().divide(10))
# 0 0.0
# 1 0.0
# 2 0.1
# 3 0.0
# dtype: float64
# Step 3
df["mean"] = df["mean"].subtract(df.groupby("mean").cumcount().divide(10))
print(df)
# Name Sum a b mean
# 0 hamm 34 2 2 2.0
# 1 jam 54 1 1 1.0
# 2 tan 36 3 1 1.9
# 3 pan 39 4 4 4.0
添加回答
舉報
0/150
提交
取消