У меня есть фрейм данных с 3 столбцами, и мне нужно получить индекс строки, который соответствует значениям с 2 столбцами.
Например, фрейм данных ниже:
Name City Country
Pietro Roma Italy
Maria Milan Italy
Pietro NY USA
В этом случае мне нужно получить индекс Pietro | Roma | Italy, который ищет только столбцы Name и City.
Я попытался выполнить приведенный ниже код, но он возвращает все строки, соответствующие 2 столбцам.
idx = np.where(dataframe[dataframe["Name"] == 'Pietro'],dataframe[dataframe["City"] == 'Roma'])
Но он возвращает массив индексов [[0,2], [0]], и мне нужно вернуть индекс 0, где у меня есть Name = 'Pietro' и City = 'Roma'
Обновлено с решением
Решение:
dataframe.index[(dataframe["Name"] == 'Pietro')&(dataframe["City"] == 'Roma')][0]