主页 / 问题 / 645307
Accepted
maestro
Asked:2020-03-28 22:57:45 +0800 CST2020-03-28 22:57:45 +0800 CST

如何在 Pandas 中找到向量的长度？

  • 772

在 DataFrame 中，每一行都是一个向量。如何计算每个向量的长度？

使用以下代码可以实现类似的行为：

import pandas as pd
import numpy as np
from scipy.spatial.distance import cdist
X = pd.DataFrame(...)
Zero = pd.DataFrame(0, index=np.arange(X.shape[0]), columns=X.columns)
L = cdist(X, Zero)

但我认为这可能更容易。该示例生成一个与 形状相同的空 DataFrame X，并计算两个 DataFrame 之间的欧几里德度量。

更新。示例初始数据集：

A       B       C
2       3       4 
2       1       2
5       4       2

结果示例：

sqrt(2^2+3^2+4^2)
sqrt(2^2+1^2+2^2)
sqrt(5^2+4^2+2^2)
python

1 个回答

  1. Best Answer

    使用sklearn.metrics.pairwise_distances - 不像scipy.spatial.distance方法，这个函数不会做不必要的（在你的情况下）计算

    源数据框的：

    In [81]: df
Out[81]:
   A  B  C
0  2  3  4
1  2  1  2
2  5  4  2

In [82]: Zero
Out[82]:
   A  B  C
0  0  0  0
1  0  0  0
2  0  0  0

    解决方案：

    In [83]: from sklearn.metrics.pairwise import paired_distances

In [84]: paired_distances(df, Zero)
Out[84]: array([ 5.38516481,  3.        ,  6.70820393])

    作为新专栏

    In [85]: df['dist'] = paired_distances(df, Zero)

In [86]: df
Out[86]:
   A  B  C      dist
0  2  3  4  5.385165
1  2  1  2  3.000000
2  5  4  2  6.708204
    • 1

相关问题