如何修改双向条形图以匹配目标示例？技术实现问询

核心修改操作 #

• Y轴标签移至左侧：设置Y轴标签和刻度的位置为左侧，确保视觉上贴近目标布局。
• 双向条形中心对接：将两组数据分别设为正负值，让X轴以0为中心对齐，实现条形在中间对接。
• 设置白色背景：把画布和坐标轴的背景色都设为白色，去除默认的灰色背景。
• 纵向布局（缩小X轴）：调整画布尺寸比例，拉长纵向高度、压缩横向宽度，实现纵向紧凑布局。
• 添加指定X轴标签：直接设置X轴的标签文本为adjusted proportion of respondents。
• 保留变量顺序与分组间隙：维持原数据的变量排序，对Volume、MAP等加粗标题对应的位置手动添加间隙。

示例代码片段 #

import matplotlib.pyplot as plt
import pandas as pd

# 假设你的原始数据结构（保持原有变量顺序）
data = pd.DataFrame({
    'Category': ['VarA', 'VarB', 'Volume', 'VarC', 'VarD', 'MAP', 'VarE'],
    'Positive': [0.3, 0.4, None, 0.2, 0.5, None, 0.1],
    'Negative': [-0.2, -0.1, None, -0.3, -0.4, None, -0.2]
})

# 创建纵向布局的画布
fig, ax = plt.subplots(figsize=(6, 10))

# 设置白色背景
fig.set_facecolor('white')
ax.set_facecolor('white')

# 绘制双向条形图
y_pos = range(len(data))
bar_width = 0.4
# 正向数据
ax.barh([y - bar_width/2 for y in y_pos], data['Positive'].fillna(0), height=bar_width, color='#1f77b4')
# 负向数据
ax.barh([y + bar_width/2 for y in y_pos], data['Negative'].fillna(0), height=bar_width, color='#ff7f0e')

# 配置Y轴标签到左侧
ax.set_yticks(y_pos)
ax.set_yticklabels(data['Category'])
ax.yaxis.set_label_position('left')
ax.yaxis.tick_left()

# 实现条形中心对接：设置对称X轴范围，将左侧轴线移到0点
x_extreme = max(data['Positive'].max(), abs(data['Negative'].min()))
ax.set_xlim(-x_extreme, x_extreme)
ax.spines['left'].set_position('zero')
ax.spines['right'].set_visible(False)
ax.spines['top'].set_visible(False)

# 添加X轴标签
ax.set_xlabel("adjusted proportion of respondents")

# 处理分组间隙与加粗标题
group_names = ['Volume', 'MAP']
for name in group_names:
    idx = data[data['Category'] == name].index[0]
    # 调整Y轴刻度位置制造间隙
    y_ticks = ax.get_yticks()
    y_ticks[idx] += 0.5
    ax.set_yticks(y_ticks)
    # 设置标题加粗
    ax.get_yticklabels()[idx].set_weight('bold')

plt.tight_layout()
plt.show()

关键细节说明 #

• 变量顺序保留：直接使用原始数据的Category列顺序生成Y轴刻度，确保顺序不变。
• 分组间隙实现：通过调整特定Y轴刻度的位置来拉开间隙，同时给对应标题设置加粗样式，匹配需求中的间隙效果。
• 中心对接优化：通过对称设置X轴范围+移动左侧轴线到0点，让正负条形精准在中心对接。

内容的提问来源于stack exchange，提问作者hulio_entredas