plot pandas dataframe via matplotlib chart and plot points on chart

Question

I am trying to plot BUY, SELL, BCLOSE and SCLOSE points only (i.e not SIT or SHRTCLS etc) from dataframe column C on the matplotlib chart that can be generated below

Eg. the at the 2nd price point in column B (5.53) as it is a BUY I am trying to plot this point on the line of the graph. The 4th price in col B is BCLOSE and this should also be added to the graph and so on.

thanks for your time.

import pandas as pd
import matplotlib.pyplot as plt

 df = pd.DataFrame()
df['A'] = ('11/06/2019','10/06/2019','9/06/2019','8/06/2019','7/06/2019','6/06/2019','5/06/2019','4/06/2019','3/06/2019','2/06/2019','1/06/2019','31/05/2019','30/05/2019')
df['B'] = (5.97,5.53,5.13,4.85,4.87,4.92,4.9,5.66,5.66,5.72,5.72,5.68,6.05)
df['C'] = ('BHODL','BUY','SIT','BCLOSE','BUY','SIT','SELL','SIT','SIT','SIT','SHRTCLS','BCLSHRT','SELL')
print(df)

ax = df.plot(title='tesing123')
ax.set_xlabel('date')
ax.set_ylabel('price')
ax.grid()
plt.show()

Vivek Kalyanarangan · Accepted Answer · 2019-08-07 11:18:10Z

1

Use -

ax = df[df['C'].isin(['BUY','SELL','BCLOSE','SCLOSE'])].plot(title='tesing123')
ax.set_xlabel('date')
ax.set_ylabel('price')
ax.grid()
plt.show()

Output

answered Aug 7, 2019 at 11:18

Vivek Kalyanarangan

9,1011 gold badge27 silver badges42 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

ds882 Over a year ago

Firstly thank you for your reply. I however must not have explained myself properly. I would like price and date to be plotted as per the dataframe, I simply however want the words BUY, SELL, BCLOSE and SCLOSE to be actually written on the point on the line on the graph when they occur

ds882 Over a year ago

2/ like a scatter plot on top of the line graph

ross · Accepted Answer · 2019-08-08 10:46:58Z

0

Try using this code:

z = df['A']
y = df['B']
n = df['C']

ax = df.plot(title='tesing123')

ax.scatter(z, y)

for i, txt in enumerate(n):
    ax.annotate(txt, (z[i], y[i]))

ax.grid()

plt.show()

edited Aug 8, 2019 at 10:46

ross

2,7842 gold badges15 silver badges22 bronze badges

answered Aug 8, 2019 at 10:12

ds882

10511 bronze badges

Collectives™ on Stack Overflow

plot pandas dataframe via matplotlib chart and plot points on chart

2 Answers 2

2 Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

2 Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related