Вnumpy Размеры результирующего массива изменяются во время выполнения. Часто возникает путаница между массивом 1d и массивом 2d с 1 столбцом. В одном случае я могу перебирать столбцы, в другом - нет.

Как вы решаете элегантно эту проблему? Чтобы не засорять мой кодif В операторах проверки на размерность я использую эту функцию:

def reshape_to_vect(ar):
    if len(ar.shape) == 1:
      return ar.reshape(ar.shape[0],1)
    return ar

Тем не менее, это чувствует себя не элегантно и дорого. Есть ли лучшее решение?