mawall
commented
4 years ago On an unrelated note: I think cmap='binary' would be nicer for voxel visualisations, especially since the default feature_vector mode is binary too and this would produce a nice figure-ground map.
In general, the single-color maps don't start from pure white, which I find slightly misleading for the density mode, as it suggests that points are uniformly distributed over the feature space.
daavoo
Don't use axes.flat for a single z-dimension. Create z_dim number of subplots for z_dim < 4. Fix if condition and hide remaining subplots for z_dim values that are not divisible by 4.