text_html_output_1.png
import pandas as pd

import seaborn as sns

wine_df = pd.read_csv('../input/winemag-data_first150k.csv')

wine_df.drop('region_1', axis=1, inplace=True)

wine_df.drop('region_2', axis=1, inplace=True)

wine_df.drop('description', axis=1, inplace=True)

wine_df.drop('designation', axis=1, inplace=True)

wine_df['reds'] = 'no'
wine_df['reds'][wine_df['variety'].str.contains('Red', case=False)] = 'yes'
wine_df['reds'][wine_df['variety'].str.contains('Cabernet', case=False)] = 'yes'
wine_df['reds'][wine_df['variety'].str.contains('Pinot Noir', case=False)] = 'yes'
wine_df['reds'][wine_df['variety'].str.contains('Syrah', case=False)] = 'yes'
wine_df['reds'][wine_df['variety'].str.contains('Malbec', case=False)] = 'yes'
wine_df['reds'][wine_df['variety'].str.contains('Sangiovese', case=False)] = 'yes'
wine_df['reds'][wine_df['variety'].str.contains('Merlot', case=False)] = 'yes'
wine_df['reds'][wine_df['variety'].str.contains('Grenache', case=False)] = 'yes'
wine_df['reds'][wine_df['variety'].str.contains('Shiraz', case=False)] = 'yes'
wine_df['reds'][wine_df['variety'].str.contains('Pinotage', case=False)] = 'yes'
wine_df['reds'][wine_df['variety'].str.contains('Monastrell', case=False)] = 'yes'
wine_df['reds'][wine_df['variety'].str.contains('Tempranillo', case=False)] = 'yes'
wine_df['reds'][wine_df['variety'].str.contains('Claret', case=False)] = 'yes'
wine_df['reds'][wine_df['variety'].str.contains('Mourvèdre', case=False)] = 'yes'
wine_df['reds'][wine_df['variety'].str.contains('Verdot', case=False)] = 'yes'
wine_df['reds'][wine_df['variety'].str.contains('Dolcetto', case=False)] = 'yes'
wine_df['reds'][wine_df['variety'].str.contains('Carmenère', case=False)] = 'yes'
wine_df['reds'][wine_df['variety'].str.contains('G-S-M', case=False)] = 'yes'