将函数逐行应用于pandas数据框

Question

我需要计算二维坐标在希尔伯特曲线上的距离。我使用了hilbertcurve包来构建自己的“hilbert”函数。这些坐标存储在一个数据框中（col_1和col_2）。如您所见，当应用于两个值（test）时，我的函数是可行的。

但是当通过apply函数逐行应用时，它就无法正常工作！这是为什么？我做错了什么？我需要一个附加列“hilbert”，其中包含从列“col_1”和“col_2”给定的x-和y-坐标的hilbert距离。

import pandas as pd
from hilbertcurve.hilbertcurve import HilbertCurve

df = pd.DataFrame({'ID': ['1', '2', '3'],
                   'col_1': [0, 2, 3],
                   'col_2': [1, 4, 5]})


def hilbert(x, y):
    n = 2
    p = 7
    hilcur = HilbertCurve(p, n)
    dist = hilcur.distance_from_coordinates([x, y])
    return dist


test = hilbert(df.col_1[2], df.col_2[2])

df["hilbert"] = df.apply(hilbert(df.col_1, df.col_2), axis=0)

上一个命令出现了错误：

The truth value of a Series is ambiguous. Use a.empty, a.bool(), a.item(), a.any() or a.all().

感谢您的帮助！

- Scrabyard

Answer 1

由于您在应用程序中使用了hilbert(df.col_1, df.col_2)，这会立即尝试使用这两列的完整pd.Series来调用您的函数，从而触发该错误。您应该这样做：

df.apply(lambda x: hilbert(x['col_1'], x['col_2']), axis=1)

这样给定的lambda函数将应用于每一行。

- Randy

---

谢谢您的回答，但是当使用您的解决方案替换最后一行时：df["hilbert"] = df.apply(lambda x: hilbert(x["col_1"], x["col_2"]), axis=0) 它会显示 KeyError: ('col_1'，'出现在索引ID处')。 - Scrabyard

事后发现，你在 axis=0 的地方应该是 axis=1 - 再试一次。 - Randy

啊，现在我明白了！谢谢！ - Scrabyard

Answer 2

你必须将轴定义为1，因为你要在行上应用函数，而不是在列上。

你可以定义一个lambda函数来仅对两行应用hilbert，如下：

df['hilbert'] = df.apply(lambda row: hilbert(row['col_1'], row['col_2']), axis=1)

- Athina Barbul