[
https://issues.apache.org/jira/browse/ARROW-2033?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=16344151#comment-16344151
]
ASF GitHub Bot commented on ARROW-2033:
---------------------------------------
wesm commented on a change in pull request #1513: ARROW-2033: [Python] Fix
pa.array() with iterator input
URL: https://github.com/apache/arrow/pull/1513#discussion_r164588063
##########
File path: cpp/src/arrow/python/builtin_convert.cc
##########
@@ -285,25 +273,44 @@ class SeqVisitor {
}
};
-Status InferArrowSize(PyObject* obj, int64_t* size) {
+// Convert *obj* to a sequence if necessary
+// Fill *size* to its length. If >= 0 on entry, *size* is an upper size
+// bound that may lead to truncation.
+Status ConvertToSequenceAndInferSize(PyObject* obj, PyObject** seq, int64_t*
size) {
if (PySequence_Check(obj)) {
- *size = static_cast<int64_t>(PySequence_Size(obj));
- } else if (PyObject_HasAttrString(obj, "__iter__")) {
+ // obj is already a sequence
+ int64_t real_size = static_cast<int64_t>(PySequence_Size(obj));
+ if (*size < 0)
+ *size = real_size;
+ else
+ *size = std::min(real_size, *size);
+ Py_INCREF(obj);
+ *seq = obj;
+ } else if (*size < 0) {
+ // unknown size, exhaust iterator
+ *seq = PySequence_List(obj);
Review comment:
I'm not sure how esoteric infinite iterators are in practice; at some point
we might want some kind of "maximum size" to prevent hanging here
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
> pa.array() doesn't work with iterators
> --------------------------------------
>
> Key: ARROW-2033
> URL: https://issues.apache.org/jira/browse/ARROW-2033
> Project: Apache Arrow
> Issue Type: Bug
> Components: Python
> Affects Versions: 0.8.0
> Reporter: Antoine Pitrou
> Assignee: Antoine Pitrou
> Priority: Minor
> Labels: pull-request-available
> Fix For: 0.9.0
>
>
> pa.array handles iterables fine, but not iterators if size isn't passed:
> {code:java}
> >>> arr = pa.array(range(5))
> >>> arr
> <pyarrow.lib.Int64Array object at 0x7f4652a05318>
> [
> 0,
> 1,
> 2,
> 3,
> 4
> ]
> >>> arr = pa.array(iter(range(5)))
> >>> arr
> <pyarrow.lib.NullArray object at 0x7f4633c1d638>
> [
> NA,
> NA,
> NA,
> NA,
> NA
> ]
> {code}
> This is because InferArrowSize() first exhausts the iterator.
--
This message was sent by Atlassian JIRA
(v7.6.3#76005)
