Python - Create a Pandas array for interval data

To create a Pandas array for interval data, use the pandas.arrays.IntervalArray() method. This creates an array structure specifically designed to handle interval data with defined start and end points.

Creating Individual Intervals

First, let's create individual Interval objects using pd.Interval() ?

import pandas as pd

# Create two Interval objects
interval1 = pd.Interval(10, 30)
interval2 = pd.Interval(30, 70)

# Display the intervals
print("Interval1...")
print(interval1)
print("Interval2...")
print(interval2)

Interval1...
(10, 30]
Interval2...
(30, 70]

Creating IntervalArray

Now we can construct an IntervalArray from these Interval objects and explore its properties ?

import pandas as pd

# Create two Interval objects
interval1 = pd.Interval(10, 30)
interval2 = pd.Interval(30, 70)

# Construct a new IntervalArray from Interval objects
array = pd.arrays.IntervalArray([interval1, interval2])

# Display the IntervalArray
print("Our IntervalArray...")
print(array)

# Getting the length of each interval
print("\nInterval lengths...")
print(array.length)

Our IntervalArray...
<IntervalArray>
[(10, 30], (30, 70]]
Length: 2, dtype: interval[int64, right]

Interval lengths...
Int64Index([20, 40], dtype='int64')

Alternative Creation Methods

You can also create IntervalArrays directly from arrays using pd.arrays.IntervalArray.from_arrays() ?

import pandas as pd

# Create IntervalArray from left and right arrays
left = [10, 30, 50]
right = [30, 70, 90]

array = pd.arrays.IntervalArray.from_arrays(left, right)
print("IntervalArray from arrays...")
print(array)
print("\nInterval midpoints...")
print(array.mid)

IntervalArray from arrays...
<IntervalArray>
[(10, 30], (30, 70], (50, 90]]
Length: 3, dtype: interval[int64, right]

Interval midpoints...
Float64Index([20.0, 50.0, 70.0], dtype='float64')

Conclusion

Use pd.arrays.IntervalArray() to create arrays for interval data. You can construct them from individual Interval objects or use from_arrays() for bulk creation from left and right boundaries.