Làm thế nào để in toàn bộ mảng NumPy mà không cần cắt bớt?

Khi tôi in một mảng numpy, tôi nhận được một đại diện cắt ngắn, nhưng tôi muốn mảng đầy đủ.

Có cách nào để làm điều này?

Ví dụ:

>>> numpy.arange(10000)
array([   0,    1,    2, ..., 9997, 9998, 9999])

>>> numpy.arange(10000).reshape(250,40)
array([[   0,    1,    2, ...,   37,   38,   39],
       [  40,   41,   42, ...,   77,   78,   79],
       [  80,   81,   82, ...,  117,  118,  119],
       ..., 
       [9880, 9881, 9882, ..., 9917, 9918, 9919],
       [9920, 9921, 9922, ..., 9957, 9958, 9959],
       [9960, 9961, 9962, ..., 9997, 9998, 9999]])

Sử dụng numpy.set_printoptions:

import sys
import numpy
numpy.set_printoptions(threshold=sys.maxsize)

import numpy as np
np.set_printoptions(threshold=np.inf)

Tôi đề nghị sử dụng np.infthay vì np.nanđược đề xuất bởi những người khác. Cả hai đều hoạt động cho mục đích của bạn, nhưng bằng cách đặt ngưỡng thành "vô cực", rõ ràng mọi người đọc mã của bạn là gì. Có một ngưỡng "không phải là số" có vẻ hơi mơ hồ đối với tôi.

Các câu trả lời trước là câu trả lời đúng, nhưng như một cách thay thế yếu hơn, bạn có thể chuyển đổi thành một danh sách:

>>> numpy.arange(100).reshape(25,4).tolist()

[[0, 1, 2, 3], [4, 5, 6, 7], [8, 9, 10, 11], [12, 13, 14, 15], [16, 17, 18, 19], [20, 21,
22, 23], [24, 25, 26, 27], [28, 29, 30, 31], [32, 33, 34, 35], [36, 37, 38, 39], [40, 41,
42, 43], [44, 45, 46, 47], [48, 49, 50, 51], [52, 53, 54, 55], [56, 57, 58, 59], [60, 61,
62, 63], [64, 65, 66, 67], [68, 69, 70, 71], [72, 73, 74, 75], [76, 77, 78, 79], [80, 81,
82, 83], [84, 85, 86, 87], [88, 89, 90, 91], [92, 93, 94, 95], [96, 97, 98, 99]]

NumPy 1.15 trở lên

Nếu bạn sử dụng NumPy 1.15 (phát hành 2018-07-23) hoặc mới hơn, bạn có thể sử dụng trình printoptionsquản lý bối cảnh:

with numpy.printoptions(threshold=numpy.inf):
    print(arr)

(tất nhiên, thay thế numpybằng npnếu đó là cách bạn nhập numpy)

Việc sử dụng trình quản lý bối cảnh ( with-block) đảm bảo rằng sau khi trình quản lý bối cảnh kết thúc, các tùy chọn in sẽ trở lại bất cứ thứ gì trước khi khối bắt đầu. Nó đảm bảo cài đặt là tạm thời và chỉ được áp dụng cho mã trong khối.

Xem numpy.printoptionstài liệu để biết chi tiết về trình quản lý bối cảnh và những đối số khác mà nó hỗ trợ.

Điều này có vẻ như bạn đang sử dụng numpy.

Nếu đó là trường hợp, bạn có thể thêm:

import numpy as np
np.set_printoptions(threshold=np.nan)

Điều đó sẽ vô hiệu hóa in góc. Để biết thêm thông tin, hãy xem Hướng dẫn NumPy này .

Đây là một cách duy nhất để làm điều này, rất hữu ích nếu bạn không muốn thay đổi cài đặt mặc định của mình:

def fullprint(*args, **kwargs):
  from pprint import pprint
  import numpy
  opt = numpy.get_printoptions()
  numpy.set_printoptions(threshold=numpy.inf)
  pprint(*args, **kwargs)
  numpy.set_printoptions(**opt)

Sử dụng một trình quản lý bối cảnh như Paul Price đã cố gắng

import numpy as np


class fullprint:
    'context manager for printing full numpy arrays'

    def __init__(self, **kwargs):
        kwargs.setdefault('threshold', np.inf)
        self.opt = kwargs

    def __enter__(self):
        self._opt = np.get_printoptions()
        np.set_printoptions(**self.opt)

    def __exit__(self, type, value, traceback):
        np.set_printoptions(**self._opt)


if __name__ == '__main__': 
    a = np.arange(1001)

    with fullprint():
        print(a)

    print(a)

    with fullprint(threshold=None, edgeitems=10):
        print(a)

numpy.savetxt

numpy.savetxt(sys.stdout, numpy.arange(10000))

hoặc nếu bạn cần một chuỗi:

import StringIO
sio = StringIO.StringIO()
numpy.savetxt(sio, numpy.arange(10000))
s = sio.getvalue()
print s

Định dạng đầu ra mặc định là:

0.000000000000000000e+00
1.000000000000000000e+00
2.000000000000000000e+00
3.000000000000000000e+00
...

và nó có thể được cấu hình với các đối số hơn nữa.

Đặc biệt lưu ý cách này cũng không hiển thị dấu ngoặc vuông và cho phép nhiều tùy chỉnh, như đã đề cập tại: Làm thế nào để in một mảng Numpy mà không có dấu ngoặc?

Đã thử nghiệm trên Python 2.7.12, numpy 1.11.1.

Đây là một sửa đổi nhỏ (loại bỏ tùy chọn để chuyển các đối số bổ sung cho câu trả lời set_printoptions)của neok .

Nó cho thấy cách bạn có thể sử dụng contextlib.contextmanagerđể dễ dàng tạo một bối cảnh như vậy với ít dòng mã hơn:

import numpy as np
from contextlib import contextmanager

@contextmanager
def show_complete_array():
    oldoptions = np.get_printoptions()
    np.set_printoptions(threshold=np.inf)
    try:
        yield
    finally:
        np.set_printoptions(**oldoptions)

Trong mã của bạn, nó có thể được sử dụng như thế này:

a = np.arange(1001)

print(a)      # shows the truncated array

with show_complete_array():
    print(a)  # shows the complete array

print(a)      # shows the truncated array (again)

Bổ sung cho câu trả lời này từ số lượng cột tối đa (cố định với numpy.set_printoptions(threshold=numpy.nan)), cũng có giới hạn các ký tự được hiển thị. Trong một số môi trường như khi gọi python từ bash (thay vì phiên tương tác), điều này có thể được khắc phục bằng cách đặt tham số linewidthnhư sau.

import numpy as np
np.set_printoptions(linewidth=2000)    # default = 75
Mat = np.arange(20000,20150).reshape(2,75)    # 150 elements (75 columns)
print(Mat)

Trong trường hợp này, cửa sổ của bạn nên giới hạn số lượng ký tự để ngắt dòng.

Đối với những người sử dụng văn bản siêu phàm và muốn xem kết quả trong cửa sổ đầu ra, bạn nên thêm tùy chọn "word_wrap": falsexây dựng vào tệp xây dựng siêu phàm [ nguồn ].

Kể từ NumPy phiên bản 1.16, để biết thêm chi tiết, xem vé GitHub 12251 .

from sys import maxsize
from numpy import set_printoptions

set_printoptions(threshold=maxsize)

Để tắt và trở về chế độ bình thường

np.set_printoptions(threshold=False)

Giả sử bạn có một mảng numpy

 arr = numpy.arange(10000).reshape(250,40)

Nếu bạn muốn in toàn bộ mảng theo cách một lần (không bật np.set_printoptions), nhưng muốn một cái gì đó đơn giản hơn (ít mã hơn) so với trình quản lý bối cảnh, chỉ cần làm

for row in arr:
     print row 

Một sửa đổi nhỏ: (vì bạn sẽ in một danh sách lớn)

import numpy as np
np.set_printoptions(threshold=np.inf, linewidth=200)

x = np.arange(1000)
print(x)

Điều này sẽ tăng số lượng ký tự trên mỗi dòng (băng thông mặc định là 75). Sử dụng bất kỳ giá trị nào bạn thích cho băng thông phù hợp với môi trường mã hóa của bạn. Điều này sẽ giúp bạn không phải trải qua số lượng lớn các dòng đầu ra bằng cách thêm nhiều ký tự trên mỗi dòng.

Bạn có thể sử dụng array2stringchức năng - tài liệu .

a = numpy.arange(10000).reshape(250,40)
print(numpy.array2string(a, threshold=numpy.nan, max_line_width=numpy.nan))
# [Big output]

Bạn sẽ không luôn muốn tất cả các mục được in, đặc biệt là cho các mảng lớn.

Một cách đơn giản để hiển thị nhiều mục hơn:

In [349]: ar
Out[349]: array([1, 1, 1, ..., 0, 0, 0])

In [350]: ar[:100]
Out[350]:
array([1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 0, 0, 1,
       1, 1, 0, 0, 0, 0, 0, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
       0, 0, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 0, 0, 0, 0,
       0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1,
       0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1])

Nó hoạt động tốt khi được cắt lát <1000 theo mặc định.

Nếu bạn có sẵn gấu trúc,

    numpy.arange(10000).reshape(250,40)
    print(pandas.DataFrame(a).to_string(header=False, index=False))

tránh tác dụng phụ của việc yêu cầu thiết lập lại numpy.set_printoptions(threshold=sys.maxsize)và bạn không nhận được numpy.array và ngoặc. Tôi thấy điều này thuận tiện cho việc bỏ một mảng rộng vào một tệp nhật ký

Nếu một mảng quá lớn để in, NumPy sẽ tự động bỏ qua phần trung tâm của mảng và chỉ in các góc: Để vô hiệu hóa hành vi này và buộc NumPy phải in toàn bộ mảng, bạn có thể thay đổi các tùy chọn in bằng cách sử dụng set_printoptions.

>>> np.set_printoptions(threshold='nan')

hoặc là

>>> np.set_printoptions(edgeitems=3,infstr='inf',
... linewidth=75, nanstr='nan', precision=8,
... suppress=False, threshold=1000, formatter=None)

Bạn cũng có thể tham khảo tài liệu numpy tài liệu numpy cho "hoặc một phần" để được trợ giúp thêm.