awkward_array_ragged_data_slicing_numpy_operations.py

1import awkward as ak
2import numpy as np
3
4# Create an Awkward Array with different lengths of lists (ragged data)
5array = ak.Array([
6    [1.1, 2.2, 3.3],
7    [],
8    [4.4, 5.5],
9    [6.6],
10    [7.7, 8.8, 9.9]
11])
12
13# Print the array
14print("Array structure:")
15print(array)
16
17# Access elements using slicing (similar to NumPy)
18print("\nFirst three lists:")
19print(array[:3])
20
21# Perform broadcasted operations
22# This adds 100 to every individual number in the nested lists
23result = array + 100
24print("\nArray after adding 100 to each element:")
25print(result)
26
27# Use NumPy functions directly on the Awkward Array
28# This calculates the square root of every element
29sqrt_array = np.sqrt(array)
30print("\nSquare root of elements:")
31print(sqrt_array)