Как рассчитать дисперсию для 2D столбцов в DataFrame

Question

У меня есть Dataframe с x, y в 2D пространстве. Похоже на долготу и широту. Я знаю, как использовать функцию var (), но не могу понять, как использовать var () с двумя разными столбцами.

Я пытаюсь вычислить дисперсию точек в 2D-пространстве.

Может Кто-нибудь помочь мне, как использовать функцию var () с 2 столбцами вместе, как оси X, Y?

Answer 1

Вы должны использовать axis=1 в своей функции df.var()

d = {'id': range(1,13),
     'lat': [62, 47, 55, 74, 31, 77, 85, 63, 42, 32, 71, 57],
     'long': [89, 87, 67, 55, 47, 72, 76, 79, 44, 92, 99, 69]}

df = pd.DataFrame(d)

df['var'] = df[['lat','long']].var(axis=1)
print(df)

Вывод:

    id  lat  long     var
0    1   62    89   364.5
1    2   47    87   800.0
2    3   55    67    72.0
3    4   74    55   180.5
4    5   31    47   128.0
5    6   77    72    12.5
6    7   85    76    40.5
7    8   63    79   128.0
8    9   42    44     2.0
9   10   32    92  1800.0
10  11   71    99   392.0
11  12   57    69    72.0