Создание подкадра данных из кадра данных - PullRequest
Новая
       23

Создание подкадра данных из кадра данных

0 голосов
/ 30 октября 2019

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

Неявный пример:

Я хочу отфильтровать Dataframe по дням, когда акции открывались с пробелами. Когда событие идентифицировано, я хочу создать график, чтобы увидеть, как акции торгуются, когда произошло это конкретное событие, поэтому мне нужны данные до и после события.

И вот я застрял, я не знаю, создавать ли дополнительный столбец вкоторый я буду использовать в качестве логического значения для того, что я ищу, или, если есть какое-то другое более элегантное решение для этого.

Благодарим вас за отзыв!

1 Ответ

0 голосов
/ 30 октября 2019 
import pandas as pd
import pandas_datareader as wb

data = wb.DataReader('spy', 'yahoo', start='1/1/1978', end='30/10/2019')
data['Change'] = data['Close'].pct_change() * 100
data['Gaps'] = ((data['Open'] - data['Close'].shift(1))/data['Close'].shift(1)) * 100
data['Performance during day'] = ((data['Close'] - data['Open'])/data['Open']) * 100
data = round(data, 2)
filtered_data = data[data['Gaps'] > 2]
print(filtered_data)
...