Сохранить значение из df в переменную
Я пытаюсь извлечь значение из фрейма данных и поместить его в переменную. Позже я запишу это значение в книгу Excel.
Сначала я запускаю SQL-запрос и сохраняю в df:
df = pd.read_sql(strSQL, conn)
Я просматриваю еще один список элементов и ищу их в df. Они связаны с помощью MMString в df и MMConcat из списка элементов, которые я просматриваю.
dftemp = df.loc[df['MMString'] == MMConcat]
Category = dftemp['CategoryName'].item()
Я получаю следующую ошибку в последней строке кода выше. ValueError: можно преобразовать только массив размером 1 в скаляр Python
В консоли отладки, когда я запускаю последнюю строку кода, но не сохраняю ее в переменной, я получаю то, что выглядит как строковое значение. Например, «Пикап».
Как я могу просто сохранить значение, которое я ищу в df, в переменной?
Ответы 1
Индексируйте по строкам столбец а также с loc, чтобы вернуть серию, затем извлеките первое значение через iat:
Category = df.loc[df['MMString'] == MMConcat, 'CategoryName'].iat[0]
В качестве альтернативы получите первое значение из представления массива NumPy:
Category = df.loc[df['MMString'] == MMConcat, 'CategoryName'].values[0]
Документы бесполезны, но pd.Series.item просто вызывает np.ndarray.item и работает только для серии с одним значением:
pd.Series([1]).item() # 1
pd.Series([1, 2]).item() # ValueError: can only convert an array of size 1
IndexError: index 0 is out of bounds for axis 0 with size 0 Это означает, что ваше состояние не соответствовало. Вы не указали, что вам нужно в этом сценарии. Вы можете использовать конструкцию try... except IndexError или использовать next(iter(df.loc[df['MMString'] == MMConcat, 'CategoryName'].values), -1), где -1 - ваше запасное значение.
Спасибо, но теперь я получаю эту ошибку: IndexError: индекс 0 выходит за пределы оси 0 с размером 0