📊 Как добавить в матрицу столбец Python: простой и эффективный способ

Для добавления столбца в матрицу в Python можно использовать библиотеку NumPy. Вот как это сделать:

import numpy as np

matrix = np.array([[1, 2, 3],
                   [4, 5, 6],
                   [7, 8, 9]])

new_column = np.array([10, 20, 30])

matrix_with_column = np.insert(matrix, 1, new_column, axis=1)

print(matrix_with_column)

Как добавить в матрицу столбец python

Добавление столбца в матрицу в Python может быть полезным, когда вам нужно расширить матрицу или ввести новые данные. В этой статье мы рассмотрим различные методы добавления столбца в матрицу с использованием примеров кода.

Метод 1: Использование спискового включения

Один из способов добавить столбец в матрицу - использовать списковое включение. Вы можете создать новый список, содержащий новые значения столбца, и затем объединить его с исходной матрицей.

# Исходная матрица
matrix = [
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9]
]

# Новый столбец
new_column = [10, 11, 12]

# Добавление столбца в матрицу
new_matrix = [row + [new_column[i]] for i, row in enumerate(matrix)]

# Вывод результата
print(new_matrix)

В этом примере мы создаем новый список `new_matrix`, который состоит из элементов исходной матрицы (`row`) и соответствующих элементов нового столбца (`new_column[i]`), добавленных в конец каждого `row`. Результат выводится на экран с помощью `print`.

Метод 2: Использование библиотеки NumPy

Если вы работаете с большими матрицами, более эффективным способом добавления столбца может быть использование библиотеки NumPy.

Установите библиотеку NumPy, если еще не установили:

pip install numpy

Затем вы можете использовать функцию `numpy.insert` для добавления столбца в матрицу.

import numpy as np

# Исходная матрица
matrix = np.array([
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9]
])

# Новый столбец
new_column = np.array([10, 11, 12])

# Добавление столбца в матрицу
new_matrix = np.insert(matrix, 2, new_column, axis=1)

# Вывод результата
print(new_matrix)

В этом примере мы используем функцию `numpy.insert`, чтобы вставить новый столбец (`new_column`) на позицию 2 (индексация начинается с 0) в исходной матрице (`matrix`). Результат сохраняется в `new_matrix` и выводится на экран с помощью `print`.

Метод 3: Использование библиотеки Pandas

Если вы работаете с табличными данными, можно использовать библиотеку Pandas для добавления столбца в матрицу.

Установите библиотеку Pandas, если еще не установили:

pip install pandas

Затем вы можете использовать метод `DataFrame.insert` для добавления столбца.

import pandas as pd

# Исходный DataFrame
df = pd.DataFrame({
    'A': [1, 2, 3],
    'B': [4, 5, 6]
})

# Новый столбец
new_column = [7, 8, 9]

# Добавление столбца в DataFrame
df.insert(2, 'C', new_column)

# Вывод результата
print(df)

В этом примере мы используем метод `insert` для добавления нового столбца `C` (`new_column`) на позицию 2 в исходный DataFrame `df`. Результат выводится на экран с помощью `print`.

Заключение

В этой статье мы рассмотрели три различных способа добавления столбца в матрицу с использованием Python. Вы можете выбрать подходящий для ваших потребностей, в зависимости от размера и типа данных матрицы, с которой вы работаете.