如何在保留非数值列的同时删除总和为0的列和行

import pandas as pd

# 加载数据集
data = pd.read_csv('data.csv')

# 检查列的数据类型，确保非数值列的数据类型正确

# 删除总和为0的列
for column in data.columns:
    if data[column].dtype != 'float64' and data[column].dtype != 'int64':
        continue
    if data[column].sum() == 0:
        data.drop(column, axis=1, inplace=True)

# 删除总和为0的行
data = data.loc[data.sum(axis=1) != 0]

# 保存修改后的数据集
data.to_csv('modified_data.csv', index=False)