python - 如何在 pandas Dataframe 中实现我自己的公式？

Question

我正在使用钻石数据集。我正在特别处理其中的两列:克拉和表。

我必须创建一个函数来获取数据帧并根据以下公式计算距离:

Python

$distance = ((carat_i - carat_j)/0.8)^2 + ((table_i - table_j)/57)^2

该函数应输出一个矩阵，其中矩阵的第 i 行和第 j 列应为第 i 个和第 j 个测量值之间的距离。

我已成功将数据帧导入为两列，如下面的代码所示。

# where df is diamonds
def exercise3(df):
    df1 = DataFrame(df)
    df2 = Dataframe(df1,
        columns=['carat','table'])

但我不确定如何实现我的公式以采取下一步

Answer 1

我用示例数据对此进行了测试，这应该有效

import pandas as pd
import numpy
data = [[5, 10], [7, 15], [10, 14]]

test_df = pd.DataFrame(data, columns = ['carat', 'table'])

def exercise3(df):
    output_array = numpy.zeros(shape=(len(df),len(df)))
    for index_i, row_i in df.iterrows():
        for index_j, row_j in df.iterrows():
            output_array[index_i,index_j] = ((row_i['carat'] - row_j['carat'])/0.8)**2 + ((row_i['table'] - row_j['table'])/57)**2
    return output_array

test = exercise3(test_df)
print(test)