Как взять значения из фрейма данных pandas и поместить их в новый список?
У меня есть фрейм данных pandas, и я хочу извлечь определенные строки, которые соответствуют условию истины, и поместить их в новый список, как мне это сделать? Пытался настроить операцию нарезки df.loc[df==101], но постоянно получаю сообщение об ошибке "Key Error = 0"
df.values[df == 101].tolist()да, вот часть фрейма данных 02.04.2018 00:00:00+00:00 Капитал(2 [ARNC]) 310 Эквити(21 [AAME]) 103 Эквити(24 [AAPL]) 311 Эквити(25 [ARNC_PR]) 101 Собственный капитал(31 [ABAX]) 206
когда когда-либо фрейм данных равен 101, я хочу вынуть строку и поместить ее в новый список
Итак, вам нужна вся строка, в которой значение столбца равно определенному значению?
Да, в этом случае это будет 101
Пожалуйста, уточните свой вопрос, каков ожидаемый результат, который вы ищете.
У меня есть фрейм данных с числом значений от -1 до 400, и я хочу вытащить и поместить в другой список все строки со значениями 101
Ответы 1
Укажите имя столбца при сравнении со значением. Вы сравниваете весь фрейм данных со 101 в df.loc[df==101]
df = pd.DataFrame(data=[['Equity(2 [ARNC])', 310],
['Equity(21 [AAME])', 103],
['Equity(24 [AAPL])', 101],
['Equity(25 [ARNC_PR])', 101],
['Equity(31 [ABAX])', 206]], columns=['Equity','Value'])
print df
Equity Value
0 Equity(2 [ARNC]) 310
1 Equity(21 [AAME]) 103
2 Equity(24 [AAPL]) 101
3 Equity(25 [ARNC_PR]) 101
4 Equity(31 [ABAX]) 206
df = df.loc[df['Value']==101]
print df
Equity Value
2 Equity(24 [AAPL]) 101
3 Equity(25 [ARNC_PR]) 101
Из этого обновленного фрейма данных вы можете получить нужный список.
newlist = df['Equity'].tolist()
print newList
['Equity(24 [AAPL])', 'Equity(25 [ARNC_PR])']
Можете ли вы привести пример фрейма данных?