Сохранение и загрузка массивов

Введение в NumPy

Izzy Weber

Curriculum Manager, DataCamp

RGB-массивы

rgb = np.array([[[255, 0, 0], [255, 0, 0], [255, 0, 0]],
                [[0, 255, 0], [0, 255, 0], [0, 255, 0]],
                [[0, 0, 255], [0, 0, 255], [0, 0, 255]]])
plt.imshow(rgb)
plt.show()

RGB-данные: красная полоса сверху, зелёная посередине, синяя снизу

RGB-массивы

Фрагмент кода с цветовой разметкой, показывающий, как смешанные цвета — розовый и жёлтый — представлены в RGB-данных Разноцветная сетка 3×3, созданная кодом из предыдущего изображения

Загрузка файлов .npy

Массивы можно сохранять в разных форматах:

.csv
.txt
.pkl
.npy

with open("logo.npy", "rb") as f:
    logo_rgb_array = np.load(f)
plt.imshow(logo_rgb_array)
plt.show()

Синий логотип NumPy на белом фоне

Анализ RGB-данных

red_array = logo_rgb_array[:, :, 0]
blue_array = logo_rgb_array[:, :, 1]
green_array = logo_rgb_array[:, :, 2]

Трёхмерный RGB-массив, разделённый на три двумерных: красный, зелёный и синий

Анализ RGB-данных

red_array[1], green_array[1], blue_array[1]

(array([255, 255, 255, ..., 255, 255, 255]),
 array([255, 255, 255, ..., 255, 255, 255]),
 array([255, 255, 255, ..., 255, 255, 255]))

Изменение RGB-данных

dark_logo_array = np.where(logo_rgb_array == 255, 50, logo_rgb_array)
plt.imshow(dark_logo_array)
plt.show()

Логотип NumPy с тёмно-серым фоном вместо белого

Сохранение массивов в файлы .npy

with open("dark_logo.npy", "wb") as f:
    np.save(f, dark_logo_array)

Если нужна help()...

help(np.unique)

Help on function unique in module numpy:
unique(ar, return_index=False, return_inverse=False, return_counts=False,
        axis=None)

Find the unique elements of an array.

Returns the sorted unique elements of an array. There are three optional
outputs in addition to the unique elements:

* the indices of the input array that give the unique values...

Снимок экрана с документацией numpy.org для np.unique()

help() для методов

help(np.ndarray.flatten)

Help on method_descriptor: flatten(...)
a.flatten(order='C')

Return a copy of the array collapsed into one dimension.

Parameters
<hr />-------
order : {'C', 'F', 'A', 'K'}, optional
    'C' means to flatten in row-major (C-style) order.
    'F' means to flatten in column-major (Fortran- ...

Давайте потренируемся!

Введение в NumPy