|
发表于 2023-4-26 10:06:29
|
显示全部楼层
您的代码存在一些问题,我将为您指出并提供修复后的代码:
1. 在for循环中,您需要使用`df.loc[i, 'gender']`而不是`df['gender']`来获取单元格的值。
2. 您需要将`if df['gender'] == 'M':`更改为`if df.loc[i, 'gender'] == 'M':`。
3. 您需要将`df['color'] = 'blue'`更改为`df.loc[i, 'color'] = 'blue'`。
修复后的代码如下:
- import pandas as pd
- # 读取 Excel 表格
- df = pd.read_excel('Python测试230426jdhbdg.xlsx')
- # 将身份证号转换为数字类型
- df['i'] = pd.to_numeric(df['i'])
- # 添加颜色列
- df['color'] = ''
- # 使用 Excel 中的条件格式将男性标记为蓝色
- for i in range(len(df)):
- if df.loc[i, 'gender'] == 'M':
- df.loc[i, 'color'] = 'blue'
- # 输出结果
- print(df)
复制代码
但是,上述代码只是在DataFrame中设置颜色列,并不能在Excel文件中实际设置单元格颜色。如果您希望在Excel文件中将男性单元格标记为蓝色,可以使用以下代码:
- import pandas as pd
- from openpyxl import load_workbook
- from openpyxl.styles import PatternFill
- # 读取 Excel 表格
- df = pd.read_excel('Python测试230426jdhbdg.xlsx')
- # 将身份证号转换为数字类型
- df['i'] = pd.to_numeric(df['i'])
- # 统计男性数量
- male_count = len(df[df['gender'] == 'M'])
- # 使用 openpyxl 设置单元格颜色
- blue_fill = PatternFill(start_color='0000FF', end_color='0000FF', fill_type='solid')
- wb = load_workbook('Python测试230426jdhbdg.xlsx')
- ws = wb.active
- # 遍历工作表,将男性标记为蓝色
- for row in ws.iter_rows(min_row=2, max_row=ws.max_row):
- if row[ws['gender'].column - 1].value == 'M':
- row[ws['gender'].column - 1].fill = blue_fill
- # 保存修改后的工作簿
- wb.save('Python测试230426jdhbdg_modified.xlsx')
- print(f"男性数量: {male_count}")
复制代码
此代码将使用`openpyxl`库操作Excel文件并设置单元格颜色。最后,会将修改后的工作簿保存为名为`Python测试230426jdhbdg_modified.xlsx`的新文件。 |
|