Значение магазина от df до переменной

Question

Я пытаюсь извлечь значение из фрейма данных и поместить его в переменную.Затем позже я запишу это значение в книгу Excel.

Сначала я запускаю SQL-запрос и сохраняю в df:

df = pd.read_sql(strSQL, conn)

Я перебираю другой список элементов и просматриваю ихв дф.Они связаны MMString в df и MMConcat из списка элементов, через которые я зацикливаюсь.

dftemp = df.loc[df['MMString'] == MMConcat]
Category = dftemp['CategoryName'].item()

Я получаю следующую ошибку в последней строке кода выше.ValueError: может преобразовывать только массив размера 1 в скаляр Python

В консоли отладки, когда я запускаю последнюю строку кода, но не сохраняю ее в переменной, я получаю то, что выглядит как строковое значение.Например, «Пикап».

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

Answer 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