使用索引作为 DataFrame 中方程的变量

Question

我有以下代码：

def linear_rank_selection(popRanked):   
selectionResults = []
n = len(popRanked)

df = pd.DataFrame(np.array(popRanked), columns=["A","B"])
df['Probability'] = 2*(n- df.index+1 )/(n*(n-1))*100

如何使用索引号作为变量来创建概率列？ 基于概率公式，而 j 是索引号。

Answer 1

当我们尝试进行除法时，我认为这里的问题在于RangeIndex 。 尝试将索引转换为 numpy：

popRanked = [(100, 50), (215, 3) , (500,4)]
n = len(popRanked)
df = pd.DataFrame(np.array(popRanked), columns=["A","B"])
df['Probability'] = 2*(n- df.index.to_numpy()+1 )/(n*(n-1))*100

print(df)
     A   B  Probability
0  100  50   133.333333
1  215   3   100.000000
2  500   4    66.666667

我们也可以使用df.index.values代替df.index.to_numpy()