В Python библиотека NumPy широко используется для эффективного выполнения числовых вычислений. Одной из распространенных задач является печать матриц, но иногда результат может быть огромным, особенно при работе с большими матрицами. В этой статье блога мы рассмотрим различные методы использования NumPy для печати матриц с усечением вывода. Мы предоставим примеры кода и объясним каждый метод в простой и разговорной форме. Итак, приступим!

Метод 1: использование np.set_printoptions
Первый метод предполагает использование функции np.set_printoptions, предоставляемой NumPy. Эта функция позволяет нам настраивать способ отображения массивов, включая матрицы. Установив параметр precision, мы можем контролировать количество отображаемых десятичных знаков. Вот пример:

import numpy as np
# Create a matrix
matrix = np.array([[1.123456789, 2.987654321],
                   [3.141592653, 4.567890123]])
# Set print options
np.set_printoptions(precision=3)
# Print the matrix
print(matrix)

Выход:

[[1.123 2.988]
 [3.142 4.568]]

Метод 2: использование форматирования строк
Другой подход заключается в использовании форматирования строк для усечения вывода каждого элемента в матрице. Мы можем добиться этого, перебирая матрицу и форматируя каждый элемент с помощью метода format. Вот пример:

import numpy as np
# Create a matrix
matrix = np.array([[1.123456789, 2.987654321],
                   [3.141592653, 4.567890123]])
# Print the matrix with truncated output
for row in matrix:
    print("[", end="")
    for element in row:
        print("{:.3f}".format(element), end=" ")
    print("]")

Выход:

[1.123 2.988 ]
[3.142 4.568 ]

Метод 3: использование np.savetxt
NumPy предоставляет функцию np.savetxt, которая позволяет нам сохранять массивы в текстовый файл. Мы можем использовать эту функцию, чтобы сохранить матрицу во временный файл, а затем прочитать и распечатать усеченное содержимое этого файла. Вот пример:

import numpy as np
# Create a matrix
matrix = np.array([[1.123456789, 2.987654321],
                   [3.141592653, 4.567890123]])
# Save matrix to temporary file
np.savetxt("temp.txt", matrix, fmt="%.3f")
# Read and print truncated contents
with open("temp.txt", "r") as file:
    contents = file.read()
    print(contents)
# Delete the temporary file
os.remove("temp.txt")

Выход:

1.123 2.988
3.142 4.568

В этой статье мы рассмотрели три различных метода печати матриц с использованием NumPy с усечением вывода. Функция np.set_printoptionsпозволяет нам настраивать точность отображения, а форматирование строк и сохранение во временный файл предоставляют альтернативные подходы. Реализуя эти методы, вы можете легко управлять выходным форматом матриц в своих программах Python. Наслаждайтесь программированием!