import numpy as np
import dedalus.public as d3

# Parameters
Nphi, Nr = 2, 1
dtype = np.float64
# Bases
coords = d3.PolarCoordinates('phi', 'r')
dist = d3.Distributor(coords, dtype=dtype)
disk = d3.DiskBasis(coords, shape=(Nphi, Nr), radius=1, dealias=1, dtype=dtype)

f = dist.Field(name='f', bases=disk)
f['c'][:2] = 1
print('f[\'c\']:', f['c'])
f['g']
for i, layout in enumerate(dist.layouts):
    print('layout %d' % i, f[layout])