Как получить столбец на основе индекса в python?

Question

У меня есть файл .csv, который я извлек в виде матрицы с именем "k":

>>> k
array([[  0.  ,   0.  ,   5.  ,  10.  ,  15.  ,  20.  ,  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.  ,
         -5.  , -10.  , -15.  , -20.  , -25.  , -30.  , -35.  , -40.  ,
        -42.  , -44.  , -46.  , -48.  , -50.  ],
       [  1.  ,   1.04,   1.02,   1.  ,   0.97,   0.95,   0.93,   0.92,
          0.92,   0.91,   0.91,   0.9 ,   0.9 ,   0.89,   0.88,   0.88,
          0.87,   0.87,   0.86,   0.85,   0.85,   0.84,   0.83,   0.82,
          0.81,   0.81,   0.8 ,   0.79,   0.77,   0.76,   0.75,   0.74,
          1.06,   1.08,   1.12,   1.14,   1.17,   1.2 ,   1.23,   1.27,
          1.28,   1.3 ,   1.32,   1.34,   1.37]

Затем я хочу извлечь все значения в столбце, который должен иметь LAT в качестве первого значения. Я пытался с этим кодом, но он извлек неправильный столбец. Кто-нибудь может мне помочь?

LAT = 42

k = np.genfromtxt(filename, delimiter = ';', filling_values = 0)

extractedData = k[:,LAT]

Answer 1

k[:, k[0] == LAT]

k[0] == LAT создает маску для выбора столбца со значением LAT в первой строке. Если вы замените его на LAT, вы получите LAT-й столбец.