Сохранять выбранный столбец как DataFrame вместо Series ⇐ Python
-
Anonymous
Сохранять выбранный столбец как DataFrame вместо Series
При выборе одного столбца из DataFrame pandas (скажем, df.iloc[:, 0], df['A'] или df.A и т. д.) результирующий вектор автоматически преобразуется в серию вместо DataFrame с одним столбцом. Однако я пишу несколько функций, которые принимают DataFrame в качестве входного аргумента. Поэтому я предпочитаю иметь дело с DataFrame с одним столбцом, а не с Series, чтобы функция могла предположить, что df.columns доступен. Прямо сейчас мне нужно явно преобразовать серию в DataFrame, используя что-то вроде pd.DataFrame(df.iloc[:, 0]). Это не похоже на самый чистый метод. Есть ли более элегантный способ напрямую индексировать DataFrame, чтобы результатом был DataFrame с одним столбцом, а не Series?
Мобильная версия