Для выполнения математических операций над столбцами в Python вы можете использовать различные библиотеки и методы в зависимости от структуры данных, с которой вы работаете. Вот некоторые распространенные методы с примерами кода:
- NumPy:
NumPy — мощная библиотека для числовых операций в Python. Вы можете выполнять математические операции над целыми столбцами, используя массивы NumPy.
import numpy as np
# Create a 2D NumPy array
data = np.array([[1, 2, 3],
[4, 5, 6],
[7, 8, 9]])
# Perform addition on columns
sum_of_columns = np.sum(data, axis=0)
print(sum_of_columns) # Output: [12 15 18]
# Perform multiplication on columns
product_of_columns = np.prod(data, axis=0)
print(product_of_columns) # Output: [28 80 162]
- Pandas.
Pandas — популярная библиотека для обработки и анализа данных. Он предоставляет объект DataFrame, который позволяет легко выполнять математические операции над столбцами.
import pandas as pd
# Create a DataFrame
data = {'A': [1, 4, 7],
'B': [2, 5, 8],
'C': [3, 6, 9]}
df = pd.DataFrame(data)
# Perform addition on columns
sum_of_columns = df.sum()
print(sum_of_columns) # Output: A 12, B 15, C 18
# Perform multiplication on columns
product_of_columns = df.prod()
print(product_of_columns) # Output: A 28, B 80, C 162
- Встроенный Python.
Если вы работаете с простыми списками или массивами, вы можете использовать встроенные функции Python для выполнения математических операций над столбцами.
# Create a list of lists
data = [[1, 2, 3],
[4, 5, 6],
[7, 8, 9]]
# Perform addition on columns
sum_of_columns = [sum(col) for col in zip(*data)]
print(sum_of_columns) # Output: [12, 15, 18]
# Perform multiplication on columns
product_of_columns = [np.prod(col) for col in zip(*data)]
print(product_of_columns) # Output: [28, 80, 162]