ndarray-npy 0.10.0

.npy and .npz file format support for ndarray
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68

"""This script generates initial files for the tests in `examples.rs`.

Afterwards, the files will be manually modified to account for slight
differences in the writer implementations. (For example, NumPy tends to write a
trailing comma in the Python dictionary literal that needs to be removed to
match the output of `py_literal`.)

"""


import numpy as np


def write_example_array(shape, dtype, order, f, dst):
    arr = np.zeros(shape, dtype=dtype, order=order)
    for i in range(arr.size):
        arr.flat[i] = f(i)
    np.save(dst, arr)


def main():
    write_example_array(
        (2, 3), dtype='<i4', order='C', f=lambda i: i,
        dst='array.npy',
    )

    SHAPE = (2, 3, 4)

    write_example_array(
        SHAPE, dtype='<f8', order='C', f=lambda i: i,
        dst='example_f64_little_endian_standard.npy',
    )
    write_example_array(
        SHAPE, dtype='<f8', order='F', f=lambda i: i,
        dst='example_f64_little_endian_fortran.npy',
    )
    write_example_array(
        SHAPE, dtype='>f8', order='C', f=lambda i: i,
        dst='example_f64_big_endian_standard.npy',
    )
    write_example_array(
        SHAPE, dtype='>f8', order='F', f=lambda i: i,
        dst='example_f64_big_endian_fortran.npy',
    )
    write_example_array(
        SHAPE, dtype='<c16', order='C', f=lambda i: i - i * 1j,
        dst='example_c64_little_endian_standard.npy',
    )
    write_example_array(
        SHAPE, dtype='<c16', order='F', f=lambda i: i - i * 1j,
        dst='example_c64_little_endian_fortran.npy',
    )
    write_example_array(
        SHAPE, dtype='>c16', order='C', f=lambda i: i - i * 1j,
        dst='example_c64_big_endian_standard.npy',
    )
    write_example_array(
        SHAPE, dtype='>c16', order='F', f=lambda i: i - i * 1j,
        dst='example_c64_big_endian_fortran.npy',
    )
    write_example_array(
        SHAPE, dtype='?', order='C', f=lambda i: (i % 5) % 2 == 0,
        dst='example_bool_standard.npy',
    )


if __name__ == '__main__':
    main()