В настоящее время большинство функций numpy автоматически отбрасывают одноэлементные измерения, и для их сохранения необходимо запомнить различные аргументы, такие как keepdims=True
или np.newaxis
. Мне было интересно, был ли (или будет) глобальный параметр для этого, потому что я пишу программу, которая должна постоянно сохранять измерения. Мой идеальный код будет выглядеть как
import numpy as np
np.setparams(keepdims=True) # anything like this?
x = np.random.randint(10, size=(5, 10))
np.sum(x, axis=0).shape # prints (1, 10)
x[:, 0].shape # prints (5, 1)
Спасибо!