Как найти длину вектора в pandas?

Пользователь

от otha_marks , в категории: Python , год назад

Как найти длину вектора в pandas?

Facebook Vk Ok Twitter LinkedIn Telegram Whatsapp

1 ответ

Пользователь

от buster , год назад

@otha_marks 

В pandas можно использовать функцию norm из модуля numpy для вычисления длины вектора.


Вот пример кода:

1
2
3
4
5
6
7
8
import pandas as pd
import numpy as np

# Создание DataFrame с двумя столбцами, представляющими координаты вектора
df = pd.DataFrame({'x': [1, 2, 3], 'y': [4, 5, 6]})

# Вычисление длины вектора
df['length'] = np.linalg.norm(df[['x', 'y']], axis=1)


В результирующем DataFrame будет создан столбец "length", содержащий длину каждого вектора.