import
pandas as pd
import
matplotlib.pyplot as plt
import
string as
str
df
=
pd.read_csv(
"car_sales.csv"
, sep
=
';'
)
df[
'car_sales_z'
]
=
df.loc[:, [
'Mercedes-Benz Cars unit sales (USA)'
]]
df[
'car_sales_z'
]
=
df[
'car_sales_z'
] .
str
.replace(
','
, '').astype(
float
)
df.drop(df.tail(
1
).index, inplace
=
True
)
for
i
in
range
(
35
):
df[
'colors'
]
=
[
'red'
if
float
(
x) <
60000
else
'green'
for
x
in
df[
'car_sales_z'
]]
df.sort_values(
'car_sales_z'
, inplace
=
True
)
df.reset_index(inplace
=
True
)
plt.figure(figsize
=
(
14
,
10
), dpi
=
80
)
plt.hlines(y
=
df.index, xmin
=
60000
, xmax
=
df.car_sales_z,
color
=
df.colors, alpha
=
0.4
, linewidth
=
5
)
plt.gca().
set
(ylabel
=
'Quarter'
, xlabel
=
'Sales'
)
plt.yticks(df.index, df.Date, fontsize
=
12
)
plt.title(
'Diverging Bars Chart Example'
, fontdict
=
{
'size'
:
20
})
plt.grid(linestyle
=
'--'
, alpha
=
0.5
)
plt.show()