text_plain_output_1.png
dropable_cols = []
for i in train_data.columns:
    if (train_data[i] == 0).all():
        dropable_cols.append(i)
train_data.drop(dropable_cols, axis=1, inplace=True)
test_data.drop(dropable_cols, axis=1, inplace=True)
print('Data shape after droping rows: ')
print('Train data shape: ', train_data.shape, 'Test data shape: ', test_data.shape)