Узнать индекс самого большого элемента
У меня есть два массива:
data = array([10654, 10437, 10604, 10735, 10936, 10279])
index = array([448, 449, 450, 451, 452, 453])
Элементы и их индексы, нужно написать программу, которая будет находить индекс большего элемента в data.
Нужно ли использовать pd.Series? Или можно каким-то другим способом это реализовать?
Ответы (1 шт):
Автор решения: MaxU
→ Ссылка
воспользуйтесь методом NDArray.argmax():
In [85]: data = np.array([10654, 10437, 10604, 10735, 10936, 10279])
In [86]: data.argmax()
Out[86]: 4
In [89]: index[data.argmax()]
Out[89]: 452
С помощью pandas.Series.idxmax():
In [90]: s = pd.Series(data, index=index)
In [91]: s
Out[91]:
448 10654
449 10437
450 10604
451 10735
452 10936
453 10279
dtype: int32
In [92]: s.idxmax()
Out[92]: 452