如何在python中基于索引获取一列?

我有一个.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]

解决方案:

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

k[0] == LAT 制作一个选择列的掩码,该掩码的值为 LAT 在第一行。如果您将其替换为 LAT 那么你就会得到LAT-th列。

本文来自投稿,不代表实战宝典立场,如若转载,请注明出处:https://www.shizhanbaodian.com/19836.html

(0)
上一篇 1天前
下一篇 1天前

相关推荐

发表评论

登录后才能评论