Vertex Coordinate Array with Values

This example is similar to the Truncated Octahedron Edges example with the addition of scalar values included to each vertex coordinate.

import numpy as np
import matplotlib.pyplot as plt
import s3dlib.surface as s3d

# 1. Define truncated octahedron geometry with values .............
a = 0.5
v = [
    [ a, 0, 1, 71], [ 0, a, 1, 49], [-a, 0, 1, 78], [ 0,-a, 1, 97], 
    [ 1, 0, a, 56], [ 0, 1, a, 56], [-1, 0, a, 42], [ 0,-1, a,112], 
    [ 1, a, 0, 77], [ a, 1, 0, 66], [-a, 1, 0, 92], [-1, a, 0, 31], 
    [-1,-a, 0, 63], [-a,-1, 0, 92], [ a,-1, 0, 43], [ 1,-a, 0, 34], 
    [ 1, 0,-a, 89], [ 0, 1,-a,103], [-1, 0,-a, 85], [ 0,-1,-a, 60], 
    [ a, 0,-1,109], [ 0, a,-1, 91], [-a, 0,-1,101], [ 0,-a,-1, 94]
]
f4 = [
    [ 0, 1, 2, 3],
    [ 4,15,16, 8], [5,9,17,10], [6,11,18,12], [ 7,13,19,14],
    [20,23,22,21]
]
f6 = [
    [ 4, 8, 9, 5, 1, 0], [ 5,10,11, 6, 2, 1], [ 6,12,13, 7, 3, 2], [ 7,14,15, 4, 0, 3],
    [16,20,21,17, 9, 8], [17,21,22,18,11,10], [18,22,23,19,13,12], [19,23,20,16,15,14]
]
f = f4 + f6

# 2. Setup and map surfaces .......................................
cmap = 'viridis'

surface = s3d.Surface3DCollection(v, f, cmap=cmap)
vertices = surface.vertices
surface.triangulate(4)
surface.map_cmap_from_vertvals()

edges = surface.initedges
edges.set_color('.6')
edges.set_linewidth(2)

# Construct figure, add surface, plot .............................
title = ['vertex values', 'interpolated face values']
fig = plt.figure(figsize=(8,4))
fig.text(0.97,0.98,str(surface), ha='right', va='top', fontsize='medium')
for i in range(2) :
    ax =fig.add_subplot(121+i, projection='3d', aspect='equal')
    ofst = 0.26 if i==0 else 0.7
    fig.text(ofst,0.1,title[i], ha='center', va='bottom', fontsize='large')
    ax.set_axis_off()
    minmax = (-.85,.85) if i==0 else (-.67,.67)  # Note: adjust for colorbar
    ax.set(xlim=minmax, ylim=minmax, zlim=minmax)
    if i==0 : 
        ax.scatter(*vertices, c=np.array(v).T[3], cmap=cmap, s=200,edgecolor='k')
        ax.add_collection3d(edges.fade(.1))
    else :
        ax.add_collection3d(surface.shade())

cbar = plt.colorbar(surface.cBar_ScalarMappable, ax=ax,  shrink=0.8 )
cbar.set_label('vertex values', rotation=270, labelpad=15)

fig.tight_layout(pad=3)
plt.show()