Mapping rpy2 objects to arbitrary python objects¶
Protocols¶
The package has a low level and a high level interface to R. The low level is
closer to R’s C API, while the high level is meant to provide more convenience
even if at the cost of performances. The low level (rpy2.rinterface
)
is not devoid of any convenience. A minimal set of Pythonic characteristics are
present, allowing rpy2 objects to behave like Python objects of similar nature
and non-rpy2 objects be sometimes usable with R functions when there is
no ambiguity about what conversion between the two systems should be.
For example, R vectors (rank-one arrays) are wrapped to rpy2 classes
implementing the methods __len_()
, __getitem__()
, __setitem__()
as defined in the sequence
protocol in Python. Python functions working with sequences can then be passed such R
objects:
import rpy2.rinterface as ri
ri.initr()
# R array of integers
r_vec = ri.IntSexpVector([1,2,3])
# enumerate() can use our r_vec
for i, elt in enumerate(r_vec):
print('r_vec[%i]: %i' % (i, elt))
rpy2 objects with compatible underlying C representations also implement
the numpy
__array_interface__
, allowing them be used in
numpy
functions without the need for data copying or conversion.
Note
Before the move to cffi
Python’s buffer protocol was also implemented
but the Python does not allow classes to define it outside of the Python C-API,
and cffi does not allow the use of the Python’s C-API.
Some rpy2 vectors will have a method memoryview()
that will return
views that implement the buffer protocol.
R functions are mapped to Python objects that implement __call__()
. They
can be called just as if they were functions.
R environments are mapped to Python objects that implement __len__()
,
__getitem__()
, __setitem__()
in the mapping protocol so elements
can be accessed similarly to in a Python dict
.
Warning
While it is technically possible to modify the way C-level R objects are shown to Python users through the rinterface level, it is not recommended. The rinterface level is quite close to R’s C API and modifying it may quickly result in segfaults.
On the other hand, the robjects-level is designed to facilitate the customization of object conversions between Python and R.
Conversion¶
The high level interface between Python in rpy2
uses a conversion system
each time an R object is represented in Python, and each time a Python objects
is passed to R (for example as a parameter to an R function). Those are the
conversion rules you’ll mostly experience when using the API in rpy2.robjects
or in the “R magic” used from ipython or jupyter.
Note
The set of active conversion rules can be customized, including within
a context (see Local conversion rules). Functions
in the rpy2.robjects
will use the active rules, but if
wanting the object with currently cactive rules rpy2.robjects.conversion.get_conversion()
must be used to fetch them.
Behind the hood, the current active conversion system is set in a
contextvars.ContextVar
. This allows the change of conversion rules to work safely
with Python context managers. However, contextvars is relatively recent and will not play
well with older Python code for multithreading. Whenever the case, the error
Conversion rules for `rpy2.robjects appear to be missing` is very likely to be encountered
when using rpy2. A workaround can be to wrap all calls to rpy2 in conversion rules’s context.
For example, to use the default converter:
import rpy2.robjects as ro with ro.default_converter.context():
# call to rpy2 here. pass
Consult the rest of the documentation for more information about conversions.
This system is designed to manage the conversion between the low level (rinterface-level) interface and an arbitrary Python-level representation those objects. py2rpy will indicate a conversion from Python-level to rinterface-level, and rpy2py from rinterface-level to Python-level.
If one wanted to turn all Python tuple
objects
into R character vectors (1D arrays of strings) before passing them to R the custom
conversion function would make an rinterface-level R objects from the Python object.
An implementation for this py2rpy function would look like:
from rpy2.rinterface import StrSexpVector
def tuple_str(tpl):
res = StrSexpVector(tpl)
return res
The conversion system is an robjects-level feature, and by default the Python-level
representations are just high-level (robjects-level) representation. However, the package contains
optional conversion rules in modules rpy2.robjects.numpy2ri
and
rpy2.robjects.pandas2ri
to convert from and to numpy
and pandas
objects respectively.
Note
Sections Numpy and Interoperability with pandas contain information about
working with rpy2 and numpy
or pandas
objects.
Converter objects¶
rpy2.robjects.conversion.Converter
objects are designed
to keep sets of conversion rules together. There can be as many instances
of that class as desired, but the one called converter in
rpy2.robjects.conversion
is the one used whenever conversion is needed.
The Converter
has 2 attributes rpy2py and py2rpy to resolve
the conversion from R (rinterface-level) to an arbitrary Python representation,
and from an arbitrary Python representation to a suitable rinterface level.
Each of those is a single dispatch as implemented in
functools.singledispatch()
. This means that a conversion function,
such as the example function tuple_str above, just has to be associated with
the class of the object to convert from. In our example, the Python class is tuple
.
Our conversion function defined above can be registered in a converter as follows:
from rpy2.robjects.conversion import Converter
seq_converter = Converter('sequence converter')
seq_converter.py2rpy.register(tuple, tuple_str)
Alternatively, the registration can be done with a decorator when the function is declared:
my_converter = rpy2.robjects.conversion.Converter()
@my_converter.py2rpy(tuple)
def tuple_str(tpl):
res = StrSexpVector(tpl)
return res
The class rpy2.robjects.conversion.Converter
can group several conversion rules
into one object. This helps will defining sets of coherent conversion rules, or
conversion domains. rpy2.robjects.numpy2ri.converter
and rpy2.rojects.pandas2ri.converter
are examples of such converters.
Sets of conversion rules can be layered on the top of one another
to create sets of combined conversion rules. To help with writing concise and
clear code, Converter
objects can be added. For example, creating a
converter that adds the rule above to the default conversion rules in rpy2
will look like:
from rpy2.robjects import default_converter
conversion_rules = default_converter + seq_converter
While a dispatch solely based on Python classes will work very well in the direction “Python to rpy2.rinterface” it will quickly show limits in the direction “rpy2.rinterface to Python”, especially when independently-developed conversions must be combined.
The issue with converting from rpy2.rinterface to Python is not working too well because rpy2.rinterface mirrors the type of R objects at the C-level (as defined in R’s C-API), but class definitions in R often sit outside of structure types found at the C level. They are just a mere attribute of the R object that contains a list class names. For example, an R data.frame is a VECSXP at C-level (that is an R list), but it has an attribute “class” that contains “data.frame”.
Note
Nothing would prevent someone to set the “class” attribute to “data.frame” to an R object of different type at C-level. For example, it is perfectly possible to write the following in R, and create an invalid data frame:
> x <- c(1, 2, 3)
> str(x)
int [1:3] 1 2 3
> class(x) <- "data.frame"
> str(x)
'data.frame': 0 obs. of 3 variables:
'data.frame' int character(0) character(0) character(0)
Warning message:
In format.data.frame(x, trim = TRUE, drop0trailing = TRUE, ...) :
corrupt data frame: columns will be truncated or padded with NAs
To allow a dispatch based name-specified classes in R, the rpy2 conversion system uses a secondary mechanism (the primary mechanism is the single dispatch-based one presented above).
Instances of rpy2.robjects.conversion.NameClassMap
can map and R class name to
a Python class. Remember that this mapping only happen within the context of an rpy2.rinterface
class though. The attribute rpy2.robjects.conversion.Converter._rpy2py_nc_name
is
a dict
where keys are rpy2.rinterface
classes to wrap C-level R objects, and
values are instances of rpy2.robjects.conversion.NameClassMap
.
For example, a conversion rule for R objects of class “lm” that are R lists at the C level (this is a real exemple - R’s linear model fit objects are just that) can be added to a converter with:
class Lm(rinterface.ListSexpVector):
# implement attributes, properties, methods to make the handling of
# the R object more convenient on the Python side
pass
clsmap = myconverter._rpy2py_nc_name[rinterface.ListSexpVector]
clsmap.update({'lm': Lm})
Local conversion rules¶
The conversion rules can be customized globally (See section Customizing the conversion) or locally in a Python with block.
Note
The use of local conversion rules is much recommended as modifying the global conversion rules can lead to wasted resources (e.g., unnecessary round-trip conversions if the code is successively passing results from calling R functions to the next R functions) or errors (conversion cannot be guaranteed to be without loss, as concepts present in either language are not always able to survive a round trip).
As an example, we show how to write an alternative to rpy2 not knowing what to do with Python tuples.
x = (1, 2, 'c')
from rpy2.robjects.packages import importr
base = importr('base')
# error here:
# NotImplementedError: Conversion 'py2rpy' not defined for objects of type '<class 'tuple'>'
res = base.paste(x, collapse="-")
This can be changed by using our converter defined above as an addition to the default conversion scheme:
from rpy2.robjects import default_converter
with conversion_rules.context():
res = base.paste(x, collapse="-")
Note
A local conversion rule can also ensure that code is robust against arbitrary changes in the conversion system made by the caller.
For example, to ensure that a function always uses rpy2’s default conversion, irrespective of what are the conversion rules defined by the caller of the code:
from rpy2.robjects import default_converter
def my_function(obj):
with default_converter.context():
# Block of code mixing Python code and calls to R functions
# interacting with the objects returned by R in the Python code.
# Within this block the conversion rules are the ones of
# `default_converter`.
pass
Code in the rpy2.robjects
will use whatever the active conversion rules are, but
there are situations where the set of active conversion rules must be accessed. Whenever
the case the conversion rules from the context manager can be named.
from rpy2.robjects import default_converter
from rpy2.robjects.conversion import get_conversion
def my_function(obj):
with default_converter.context() as local_converter:
# `local_converter` is a rpy2.robjects.conversion.Converter
# object.
pass
The converter returned by :meth:`rpy2.robjects.conversion.Converter.context` is
a copy of the rules for the context.
.. code-block:: python
with default_converter.context() as local_converter:
# Conversion objects are not the same.
assert local_converter != default_converter
assert cv.py2rpy.registry != default_converter.py2rpy
assert cv.rpy2py.registry != default_converter.rpy2py
# The convertion rules are identical though.
assert dict(cv.py2rpy.registry) == dict(default_converter.py2rpy.registry)
assert dict(cv.rpy2py.registry) == dict(default_converter.rpy2py.registry)
Customizing the conversion¶
As an example, let’s assume that one want to return atomic values whenever an R numerical vector is of length one. This is only a matter of writing a new function rpy2py that handles this, as shown below:
import rpy2.robjects as robjects
from rpy2.rinterface import SexpVector
@robjects.conversion.rpy2py.register(SexpVector)
def my_rpy2py(obj):
if len(obj) == 1:
obj = obj[0]
return obj
Then we can test it with:
>>> pi = robjects.r.pi
>>> type(pi)
<type 'float'>
At the time of writing singledispath()
does not provide a way to unregister.
Removing the additional conversion rule without restarting Python is left as an
exercise for the reader.
Note
Customizing the conversion of S4 classes should preferably done using a separate dedicated system.
The system is rather simple and can easily be described with an example.
import rpy2.robjects as robjects
from rpy2.robjects.packages import importr
class LMER(robjects.RS4):
"""Custom class."""
pass
lme4 = importr('lme4')
res = robjects.r('lmer(Reaction ~ Days + (Days | Subject), sleepstudy)')
# Map the R/S4 class 'lmerMod' to our Python class LMER.
with robjects.conversion.converter.rclass_map_context(
rinterface.rinterface.SexpS4,
{'lmerMod': LMER}
):
res2 = robjects.r('lmer(Reaction ~ Days + (Days | Subject), sleepstudy)')
When running the example above, res is an instance of class
rpy2.robjects.methods.RS4
,
which is the default mapping for R S4 instances, while res2 is an instance of our
custom class LMER.
The class mapping is using the hierarchy of R/S4-defined classes and tries to find the first matching Python-defined class. For example, the R/S4 class lmerMod has a parent class merMod (defined in R S4). Let run the following example after the previous one.
class MER(robjects.RS4):
"""Custom class."""
pass
with robjects.conversion.converter.rclass_map_context(
rinterface.rinterface.SexpS4,
{'merMod': MER}
):
res3 = robjects.r('lmer(Reaction ~ Days + (Days | Subject), sleepstudy)')
with robjects.conversion.converter.rclass_map_context(
rinterface.rinterface.SexpS4,
{'lmerMod': LMER,
'merMod': MER}):
res4 = robjects.r('lmer(Reaction ~ Days + (Days | Subject), sleepstudy)')
res3 will be a MER instance: there is no mapping for the R/S4 class lmerMod but there is a mapping for its R/S4 parent merMod. res4 will be an LMER instance.