在 Python 中实现 if 条件分组并控制分组间距,通常需要结合 条件判断 和 数据可视化/排版技巧。以下是针对不同场景的解决方案:
一、数据分组场景(如 Pandas 数据处理)
1.按条件分组并计算间距
import pandas as pd
# 示例数据
data = {'value': [12, 45, 7, 23, 36, 9, 18],
'group': ['A', 'B', 'A', 'C', 'B', 'A', 'C']}
df = pd.DataFrame(data)
# 按条件分组并添加间距列
df['spacing'] = df['group'].apply(
lambda x: 10 if x == 'A' else (20 if x == 'B' else 30)
)
print(df)
输出:
value group spacing
0 12 A 10
1 45 B 20
2 7 A 10
3 23 C 30
4 36 B 20
5 9 A 10
6 18 C 30
2.分组后聚合(带间距调整)
# 按组计算均值,并附加间距元数据
group_stats = df.groupby('group').agg({
'value': 'mean',
'spacing': 'first' # 取每组的第一个间距值
})
print(group_stats)
二、可视化分组间距(Matplotlib/Seaborn)
1.柱状图分组间距控制
import matplotlib.pyplot as plt
# 分组数据
groups = ['A', 'B', 'C']
values = [df[df['group'] == g]['value'].mean() for g in groups]
spacings = [10, 20, 30] # 自定义各组间距
# 绘制带间距的分组柱状图
fig, ax = plt.subplots()
bars = ax.bar(groups, values, width=0.6)
# 调整分组间距(通过调整x轴位置)
for i, (bar, spacing) in enumerate(zip(bars, spacings)):
bar.set_x(i * (1 + spacing/100)) # 按比例调整位置
plt.show()
2.Seaborn 的hue分组间距
import seaborn as sns
# 创建示例数据(带子分组)
tips = sns.load_dataset("tips")
tips['spacing_group'] = tips['day'].apply(
lambda x: 0.1 if x == 'Thur' else 0.3)
# 绘制带间距的分组箱线图
sns.boxplot(
x="day",
y="total_bill",
hue="sex",
data=tips,
gap=tips['spacing_group'].mean() # 控制子组间距
)
三、纯 Python 列表分组(基础实现)
data = [12, 45, 7, 23, 36, 9, 18]
groups = ['A', 'B', 'A', 'C', 'B', 'A', 'C']
# 分组并添加间距
group_dict = {}
for val, grp in zip(data, groups):
spacing = 10 if grp == 'A' else (20 if grp == 'B' else 30)
if grp not in group_dict:
group_dict[grp] = {'values': [], 'spacing': spacing}
group_dict[grp]['values'].append(val)
print(group_dict)
输出:
{
'A': {'values': [12, 7, 9], 'spacing': 10},
'B': {'values': [45, 36], 'spacing': 20},
'C': {'values': [23, 18], 'spacing': 30}
}
四、HTML/CSS 分组间距(Web 应用)
from IPython.display import HTML
# 生成带间距的分组HTML表格
groups = {'A': [1, 2], 'B': [3, 4], 'C': [5, 6]}
spacings = {'A': '20px', 'B': '40px', 'C': '60px'}
html = "<div>"
for grp, items in groups.items():
html += f"""
<div style='margin-bottom: {spacings[grp]}; border: 1px solid #ccc; padding: 10px'>
<h3>Group {grp}</h3>
<ul>
{"".join(f"<li>{x}</li>" for x in items)}
</ul>
</div>
"""
html += "</div>"
HTML(html)
五、关键注意事项
间距单位一致性:
- 数据操作:使用数值型间距(如 10, 20, 30)
- 可视化:使用相对单位(如 0.1, 0.3)
- 前端:使用 CSS 单位(px/em/rem)
动态间距计算:
# 根据组大小自动调整间距
df['dynamic_spacing'] = df.groupby('group')['group'].transform('count') * 5
性能优化:
- 大数据集避免循环,优先使用 Pandas/Numpy 向量化操作
六、完整案例:销售数据分组报表
import pandas as pd
# 生成示例数据
sales = pd.DataFrame({
'region': ['North']*3 + ['South']*2 + ['East']*4,
'sales': [120, 150, 90, 200, 180, 70, 85, 110, 95]
})
# 定义分组间距规则
spacing_rules = {'North': 15, 'South': 25, 'East': 10}
# 添加间距并生成报表
report = (
sales.groupby('region')
.agg(total_sales=('sales', 'sum'))
.assign(spacing=lambda x: x.index.map(spacing_rules))
)
print(report)
输出:
total_sales spacing
region
East 360 10
North 360 15
South 380 25
通过以上方法,你可以灵活实现:
数据分组 + 自定义间距
可视化控制 + 动态调整
跨平台兼容(数据/Web/可视化)