结合列值在牛市绘制栅格堆叠条形图

Question

我想绘制一个网格叠加条形图，如栅格堆叠条形图示例中所示。

我有这个数据集：

pd.DataFrame({
    
    'storage': ['dev01', 'dev01', 'dev01', 'dev02', 'dev02', 'dev03'],
    'project': ['omega', 'alpha', 'beta', 'omega', 'beta', 'alpha'],
    'read': [3, 0, 0, 114, 27, 82],
    'write': [70, 0, 0, 45, 655, 203],
    'read-write': [313, 322, 45, 89, 90, 12]
    
})

  storage project  read  write  read-write
0   dev01   omega     3     70         313
1   dev01   alpha     0      0         322
2   dev01    beta     0      0          45
3   dev02   omega   114     45          89
4   dev02    beta    27    655          90
5   dev03   alpha    82    203          12

我无法理解的是如何将read、write、read-write列指定为Altair的颜色/值。

Answer 1

您的数据是宽格式的，必须转换为长型，以便在牛尾编码中使用.有关更多信息，请参见牛市文档中的长型数据与宽格式数据。

可以通过使用pd.melt修改Pandas中的输入数据来解决这一问题，但是使用Altair的褶皱变换在图表规范中进行这种整形通常更方便。例如：

import pandas as pd
import altair as alt

df = pd.DataFrame({
    'storage': ['dev01', 'dev01', 'dev01', 'dev02', 'dev02', 'dev03'],
    'project': ['omega', 'alpha', 'beta', 'omega', 'beta', 'alpha'],
    'read': [3, 0, 0, 114, 27, 82],
    'write': [70, 0, 0, 45, 655, 203],
    'read-write': [313, 322, 45, 89, 90, 12]
})

alt.Chart(df).transform_fold(
    ['read', 'write', 'read-write'],
    as_=['mode', 'value']
).mark_bar().encode(
    x='value:Q',
    y='project:N',
    column='storage:N',
    color='mode:N'
).properties(
    width=200
)

​

​

Answer 2

需要将所需列melt为新列：

# assuming your DataFrame is assigned to `df`

cols_to_melt = ['read', 'write', 'read-write']
cols_to_keep = df.columns.difference(cols_to_melt)

df = df.melt(cols_to_keep, cols_to_melt, 'mode')

所以你得到了以下信息：

   project storage        mode  value
0    omega   dev01        read      3
1    alpha   dev01        read      0
2     beta   dev01        read      0
3    omega   dev02        read    114
4     beta   dev02        read     27
5    alpha   dev03        read     82
6    omega   dev01       write     70
7    alpha   dev01       write      0
8     beta   dev01       write      0
9    omega   dev02       write     45
10    beta   dev02       write    655
11   alpha   dev03       write    203
12   omega   dev01  read-write    313
13   alpha   dev01  read-write    322
14    beta   dev01  read-write     45
15   omega   dev02  read-write     89
16    beta   dev02  read-write     90
17   alpha   dev03  read-write     12

然后在牛郎星片段中，而不是在color='site'中，使用color='mode'。