numpyで中央値を求めたい

Question

numpyで中央値を求めたい

positions = ['GK', 'M', 'A', 'D', ...]
heights = [191, 184, 185, 180, ...]

上記の前提があった場合に、GK（ゴールキーパー）の中央値を求めたいです。

エラー箇所は、「gk_heights = np_heights(np_positions.index('GK'))」という部分ですが、
positionsに対応したheightsの値は、どのように取り出せるでしょうか。

発生している問題・エラーメッセージ

Traceback (most recent call last):
  File "<stdin>", line 12, in <module>
    gk_heights = np_heights(np_positions.index('GK'))
AttributeError: 'numpy.ndarray' object has no attribute 'index'

該当のソースコード

# heights and positions are available as lists

# Import numpy
import numpy as np

# Convert positions and heights to numpy arrays: np_positions, np_heights
np_positions = np.array(positions)
np_heights = np.array(heights)

# Heights of the goalkeepers: gk_heights←ここが分からない
gk_heights = np_heights(np_positions.index('GK'))

# Heights of the other players: other_heights
other_heights = np_heights(np_positions.index('M', 'A', 'D'))

# Print out the median height of goalkeepers. Replace 'None'
print("Median height of goalkeepers: " + str(np.median(gk_heights)))

# Print out the median height of other players. Replace 'None'
print("Median height of other players: " + str(np.median(other_heights)))

Answer 1

>>> np_heights[np_positions == 'GK']
array([191])
>>> np_heights[np_positions == 'GK'][0]
191

numpyを使わない方法なら・・・

gk_heights = heights[positions.index('GK')]

あるいは

gk_heights = dict(zip(positions, heights))['GK']

Comments

[pablo_2019]

ありがとうございます！！
解決できました、、、。
本当に助かりました、泣