Срезы в двумерном массиве
В NumPy срезы позволяют вам обращаться к строкам и столбцам двумерного массива, а также заменять их. Давайте рассмотрим, как это сделать.
1. Обращение к строкам
Чтобы обратиться к строкам массива, вы можете использовать срезы.
import numpy as np
# Создаем двумерный массив 3x3
array = np.array([[1, 2, 3],
[4, 5, 6],
[7, 8, 9]])
# Получаем первые две строки
first_two_rows = array[0:2]
print("Первые две строки:")
print(first_two_rows)
# Вывод:
# [[1 2 3]
# [4 5 6]]
2. Обращение к столбцам
Чтобы обратиться к столбцам, вы можете использовать срезы с запятой. Например, чтобы получить первый столбец:
# Получаем первые два столбца
first_two_columns = array[:, 0:2]
print("Первые два столбца:")
print(first_two_columns)
# Вывод:
# [[1 2]
# [4 5]
# [7 8]]
3. Замена строк и столбцов
Вы также можете заменять строки и столбцы, используя срезы.
Замена строки
Например, заменим вторую строку:
# Заменяем вторую строку
array[1] = [10, 11, 12]
print("После замены второй строки:")
print(array)
# Вывод:
# [[ 1 2 3]
# [10 11 12]
# [ 7 8 9]]
Замена столбца
Чтобы заменить столбец, используйте срезы с запятой:
# Заменяем первый столбец
array[:, 0] = [100, 200, 300]
print("После замены первого столбца:")
print(array)
# Вывод:
# [[100 2 3]
# [200 11 12]
# [300 8 9]]