Python - Create a Pandas array for interval data

To create a Pandas array for interval data, use the pandas.arrays.IntervalArray() method. At first, import the required libraries −

import pandas as pd

Create two Interval objects −

interval1 = pd.Interval(10, 30)
interval2 = pd.Interval(30, 70)

Display the intervals −

print("Interval1...\n",interval1)
print("Interval2...\n",interval2)

Construct a new IntervalArray from Interval objects −

array = pd.arrays.IntervalArray([interval1,interval2])

Example

Following is the code −

import pandas as pd

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

# display the intervals
print("Interval1...\n",interval1)
print("Interval2...\n",interval2)

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

# Display the IntervalArray
print("\nOur IntervalArray...\n",array)

# Getting the length of IntervalArray
# Returns an Index with entries denoting the length of each Interval in the IntervalArray
print("\nOur IntervalArray length...\n",array.length)

Output

This will produce the following code −

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

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

Our IntervalArray length...
Int64Index([20, 40], dtype='int64')