Как получить значения в массиве, используя другой массив с индексами значений?

У меня есть массив с плавающей запятой (размер 9300) и еще один массив целых чисел (размер 2600) с индексами первого массива. Я пытаюсь получить значения в первом массиве на основе индекса во втором массиве. Пример:

index   1st_array
0       12.00
1       3.89 
2       24.20
3       34.60

index  2nd_array
0       0
1       2

Выход:

index      3nd_array
0          12.00
1          24.20

Обновлено: на самом деле это пустые массивы:

a1 = np.array([12.00, 3.89, 24.20, 34.60])
a2 = np.array([0, 2])
Анализ настроения постов в Twitter с помощью Python, Tweepy и Flair
Анализ настроения постов в Twitter с помощью Python, Tweepy и Flair
Анализ настроения текстовых сообщений может быть настолько сложным или простым, насколько вы его сделаете. Как и в любом ML-проекте, вы можете выбрать...
7 лайфхаков для начинающих Python-программистов
7 лайфхаков для начинающих Python-программистов
В этой статье мы расскажем о хитростях и советах по Python, которые должны быть известны разработчику Python.
Установка Apache Cassandra на Mac OS
Установка Apache Cassandra на Mac OS
Это краткое руководство по установке Apache Cassandra.
Сертификатная программа "Кванты Python": Бэктестер ансамблевых методов на основе ООП
Сертификатная программа "Кванты Python": Бэктестер ансамблевых методов на основе ООП
В одном из недавних постов я рассказал о том, как я использую навыки количественных исследований, которые я совершенствую в рамках программы TPQ...
Создание персонального файлового хранилища
Создание персонального файлового хранилища
Вы когда-нибудь хотели поделиться с кем-то файлом, но он содержал конфиденциальную информацию? Многие думают, что электронная почта безопасна, но это...
Создание приборной панели для анализа данных на GCP - часть I
Создание приборной панели для анализа данных на GCP - часть I
Недавно я столкнулся с интересной бизнес-задачей - визуализацией сбоев в цепочке поставок лекарств, которую могут просматривать врачи и...
2
0
42
2
Перейти к ответу Данный вопрос помечен как решенный

Ответы 2

import numpy as np
arr1 = np.array([12.00, 3.89, 24.20, 34.60])
arr2 = np.array([0, 2])
print(arr1[arr2])

отпечатки

[12.  24.2]
Ответ принят как подходящий

Если у вас две серии s1 и s2, используйте:

s1.loc[s2]

Пример:

s1 = pd.Series(['a', 'b', 'c', 'd'])
s2 = pd.Series([0, 2])

s1.loc[s2]
0    a
2    c
dtype: object

Для пустых массивов

a1[a2]

Пример:

a1 = np.array(['a', 'b', 'c', 'd'])
a2 = np.array([0, 2])

a1[a2]
array(['a', 'c'], dtype='<U1')

Для фреймов данных:

df1.set_index('index').loc[df2['2nd_array']].reset_index()

NB. set_index/reset_index требуются только в том случае, если «индекс» является столбцом

Пример:

df1 = pd.DataFrame({'index': [0, 1, 2, 3], 
                    '1st_array': [12.0, 3.89, 24.2, 34.6]})

df2 = pd.DataFrame({'index': [0, 1],
                    '2nd_array': [0, 2]})

(df1
 .set_index('index')
 .loc[df2['2nd_array']]
 .reset_index()
 )

   index  1st_array
0      0       12.0
1      2       24.2

@ Фред, так какой у тебя был случай?

mozway 23.04.2022 07:54

Для массивов numpy, но один для фрейма данных в конечном итоге полезен и для чего-то еще.

Fred 23.04.2022 08:02

Другие вопросы по теме