pyarrow.FixedShapeTensorType#

classpyarrow.FixedShapeTensorType#

Bases:BaseExtensionType

Concrete class for fixed shape tensor extension type.

Examples

Create an instance of fixed shape tensor extension type:

>>>importpyarrowaspa>>>pa.fixed_shape_tensor(pa.int32(),[2,2])FixedShapeTensorType(extension<arrow.fixed_shape_tensor[value_type=int32, shape=[2,2]]>)

Create an instance of fixed shape tensor extension type withpermutation:

>>>tensor_type=pa.fixed_shape_tensor(pa.int8(),(2,2,3),...permutation=[0,2,1])>>>tensor_type.permutation[0, 2, 1]
__init__(*args,**kwargs)#

Methods

__init__(*args, **kwargs)

equals(self, other, *[, check_metadata])

Return true if type is equivalent to passed value.

field(self, i)

to_pandas_dtype(self)

Return the equivalent NumPy / Pandas dtype.

wrap_array(self, storage)

Wrap the given storage array as an extension array.

Attributes

bit_width

The bit width of the extension type.

byte_width

The byte width of the extension type.

dim_names

Explicit names of the dimensions.

extension_name

The extension type name.

has_variadic_buffers

If True, the number of expected buffers is only lower-bounded by num_buffers.

id

num_buffers

Number of data buffers required to construct Array type excluding children.

num_fields

The number of child fields.

permutation

Indices of the dimensions ordering.

shape

Shape of the tensors.

storage_type

The underlying storage type.

value_type

Data type of an individual tensor.

bit_width#

The bit width of the extension type.

byte_width#

The byte width of the extension type.

dim_names#

Explicit names of the dimensions.

equals(self,other,*,check_metadata=False)#

Return true if type is equivalent to passed value.

Parameters:
otherDataType orstrconvertible toDataType
check_metadatabool

Whether nested Field metadata equality should be checked as well.

Returns:
is_equalbool

Examples

>>>importpyarrowaspa>>>pa.int64().equals(pa.string())False>>>pa.int64().equals(pa.int64())True
extension_name#

The extension type name.

field(self,i)Field#
Parameters:
iint
Returns:
pyarrow.Field
has_variadic_buffers#

If True, the number of expected buffers is onlylower-bounded by num_buffers.

Examples

>>>importpyarrowaspa>>>pa.int64().has_variadic_buffersFalse>>>pa.string_view().has_variadic_buffersTrue
id#
num_buffers#

Number of data buffers required to construct Array typeexcluding children.

Examples

>>>importpyarrowaspa>>>pa.int64().num_buffers2>>>pa.string().num_buffers3
num_fields#

The number of child fields.

Examples

>>>importpyarrowaspa>>>pa.int64()DataType(int64)>>>pa.int64().num_fields0>>>pa.list_(pa.string())ListType(list<item: string>)>>>pa.list_(pa.string()).num_fields1>>>struct=pa.struct({'x':pa.int32(),'y':pa.string()})>>>struct.num_fields2
permutation#

Indices of the dimensions ordering.

shape#

Shape of the tensors.

storage_type#

The underlying storage type.

to_pandas_dtype(self)#

Return the equivalent NumPy / Pandas dtype.

Examples

>>>importpyarrowaspa>>>pa.int64().to_pandas_dtype()<class 'numpy.int64'>
value_type#

Data type of an individual tensor.

wrap_array(self,storage)#

Wrap the given storage array as an extension array.

Parameters:
storageArray orChunkedArray
Returns:
arrayArray orChunkedArray

Extension array wrapping the storage array