Reputation: 8025
I have a working code that displays a panda dataframe as 2 line graphs in a chart. I also have a dataframe that displays a bar graph on the same chart. For the 2 dataframes, i have date for the x-axis. Because the two dataframes have dates, my axis end up just having integers (1,2,3,4,5,6...) instead of the date.
I thought this line df1 = df.set_index(['date'])
specifies what i want as the x-axis already and when i dont plot the bar graph on the chart, the dates show up nicely, but when i do plot the bar graph, the integers show up on the axis instead.
My 2 dataframes:
df1:
date line1 line2
2015-01-01 15.00 23.00
2015-02-01 18.00 10.00
df2:
date quant
2015-01-01 500
2015-02-01 600
My code:
df1 =pd.DataFrame(result, columns =[ 'date','line1', 'line2'])
df1 = df.set_index(['date'])
df2 =pd.DataFrame(quantity, columns =[ 'quant','date'])
fig = plt.figure()
ax = fig.add_subplot(111)
ax2=ax.twinx()
ax.set_ylim(0,100)
ax2.set_ylim(0,2100)
df1.line1.plot( color = 'red', ax = ax)
df1.line2.plot( color = 'blue', ax = ax)
df2.["quant"].plot(kind = 'bar', ax =ax2, width =0.4)
plt.show()
df1:
<class 'pandas.core.frame.DataFrame'>
Int64Index: 12 entries, 0 to 11
Data columns (total 3 columns):
date 12 non-null object
line1 12 non-null float64
line2 12 non-null float64
dtypes: float64(2), object(1)
memory usage: 384.0+ bytes
None
df2
<class 'pandas.core.frame.DataFrame'>
Int64Index: 11 entries, 0 to 10
Data columns (total 2 columns):
quant 11 non-null int64
date 11 non-null object
dtypes: int64(1), object(1)
memory usage: 264.0+ bytes
None
Upvotes: 0
Views: 9676
Reputation: 24742
You can just use ax.plot(df1.date, df1.line1)
and matplotlib.pyplot
will automatically take care of the date.
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
# your data
# ===================================
np.random.seed(0)
df1 = pd.DataFrame(dict(date=pd.date_range('2015-01-01', periods=12, freq='MS'), line1=np.random.randint(10, 30, 12), line2=np.random.randint(20, 25, 12)))
Out[64]:
date line1 line2
0 2015-01-01 22 22
1 2015-02-01 25 21
2 2015-03-01 10 20
3 2015-04-01 13 21
4 2015-05-01 13 21
5 2015-06-01 17 20
6 2015-07-01 19 21
7 2015-08-01 29 24
8 2015-09-01 28 23
9 2015-10-01 14 20
10 2015-11-01 16 23
11 2015-12-01 22 20
df2 = pd.DataFrame(dict(date=pd.date_range('2015-01-01', periods=12, freq='MS'), quant=100*np.random.randint(3, 10, 12)))
Out[66]:
date quant
0 2015-01-01 500
1 2015-02-01 600
2 2015-03-01 300
3 2015-04-01 400
4 2015-05-01 600
5 2015-06-01 800
6 2015-07-01 600
7 2015-08-01 600
8 2015-09-01 900
9 2015-10-01 300
10 2015-11-01 400
11 2015-12-01 400
# plotting
# ===================================
fig, ax = plt.subplots(figsize=(10, 8))
ax.plot(df1.date, df1.line1, label='line1', c='r')
ax.plot(df1.date, df1.line2, label='line2', c='b')
ax2 = ax.twinx()
ax2.set_ylabel('quant')
ax2.bar(df2.date, df2.quant, width=20, alpha=0.1, color='g', label='quant')
ax.legend(loc='best')
ax.set_xticks(ax.get_xticks()[::2])
Follow-up (Updates):
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
# your data
# ===================================
np.random.seed(0)
df1 = pd.DataFrame(dict(date=pd.date_range('2015-01-01', periods=12, freq='MS'), line1=np.random.randint(10, 30, 12), line2=np.random.randint(20, 25, 12))).set_index('date')
df2 = pd.DataFrame(dict(date=pd.date_range('2015-01-01', periods=12, freq='MS'), quant=100*np.random.randint(3, 10, 12))).set_index('date')
df2 = df2.drop(df2.index[4])
print(df1)
print(df2)
line1 line2
date
2015-01-01 22 22
2015-02-01 25 21
2015-03-01 10 20
2015-04-01 13 21
2015-05-01 13 21
2015-06-01 17 20
2015-07-01 19 21
2015-08-01 29 24
2015-09-01 28 23
2015-10-01 14 20
2015-11-01 16 23
2015-12-01 22 20
quant
date
2015-01-01 500
2015-02-01 600
2015-03-01 300
2015-04-01 400
2015-06-01 800
2015-07-01 600
2015-08-01 600
2015-09-01 900
2015-10-01 300
2015-11-01 400
2015-12-01 400
# plotting
# ===================================
fig, ax = plt.subplots(figsize=(10, 8))
ax.plot(df1.index, df1.line1, label='line1', c='r')
ax.plot(df1.index, df1.line2, label='line2', c='b')
ax2 = ax.twinx()
ax2.set_ylabel('quant')
ax2.bar(df2.index, df2.quant, width=20, alpha=0.1, color='g', label='quant')
ax.legend(loc='best')
ax.set_xticks(ax.get_xticks()[::2])
Upvotes: 1