Это лучший способ добавить дополнительное измерение в массив numpy в одной строке кода?
Если k является массивом numpy произвольной формы, поэтому k.shape = (s1, s2, s3, ..., sn), и я хочу изменить его так, чтобы k.shape стал (s1, s2, ..., sn, 1), это лучший способ сделать это в одной строке?
k.reshape(*(list(k.shape) + [1])
Ответ 1
Это проще:
k.reshape(k.shape + (1,))
Но если вы хотите добавить пустое измерение в конец, вы должны использовать numpy.newaxis: