lặp trên sản phẩm Cartesian của vectơ

Question

tôi đã vòng lặp lồng nhau sau:

for (x in xs) { 
    for (y in ys) { 
     # Do something with x and y 
    } 
} 

Mà tôi muốn flatten vì vậy tôi nghĩ đến việc xây dựng một sản phẩm Descartes của hai vectơ xs và ys và lặp lại qua kết quả . Trong Python, đây sẽ là tầm thường:

for xy in product(xs, ys): 
    # x, y = xy[0], xy[1] 

Nhưng trong R, tương đương đơn giản nhất tôi đã tìm thấy trông khó khăn:

xys <- expand.grid(xs, ys) 
for (i in 1 : nrow(xys)) { 
    xy <- as.vector(xys[i, ]) 
    # x <- xy[1], y <- xy[2] 
} 

Chắc chắn phải có một cách tốt hơn, đúng không? (Để làm rõ, tôi không muốn lặp qua một chỉ số ... Tôi nghĩ rằng phải có cách để trực tiếp lặp qua các bộ dữ liệu trong sản phẩm.)

Answer 1

Bạn có thể sử dụng chức năng apply để áp dụng một chức năng cho mỗi hàng trong khung dữ liệu của bạn. Chỉ cần thay thế "your function" bằng chức năng thực tế của bạn.

# example data 
xs <- rnorm(10) 
ys <- rnorm(10)  

apply(expand.grid(xs, ys), 1, FUN = function(x) {"your function"}) 

Đây là ví dụ rất cơ bản. Ở đây, tổng của cả hai giá trị liên tiếp được tính:

apply(expand.grid(xs, ys), 1, FUN = function(x) {x[1] + x[2]}) 

---

đây là một biến thể sử dụng đối số tên (xs, ys) thay vì chỉ số (x[1], x[2]):

myfun <- function(xs, ys) xs + ys 
arguments <- expand.grid(xs = rnorm(10), ys = rnorm(10)) 
apply(arguments, 1, function(x)do.call(myfun, as.list(x))) 

Answer 2

R có một mô hình khác với Python, vì vậy đừng ' Tôi hy vọng nó có máy phát điện hoặc bộ tuples - chúng tôi có vectơ và chỉ số cho điều đó.

Bằng cách này, để ánh xạ một chức năng trên một sản phẩm Descartes chỉ cần gọi

outer(xs,ys,function(x,y) ...) 

và undim kết quả nếu bạn muốn.

EDIT: Trong trường hợp xs hoặc ys là một cái gì đó phức tạp hơn vectơ cơ sở, một lựa chọn là sử dụng các chỉ số, tức là

outer(seq(a=xs),seq(a=ys),function(xi,yi) ... xs[[xi]]/ys[xi,]/etc. ...) 

hoặc bản đồ một chức năng trên một sản phẩm hand-crafted chút bằng mapply

mapply(function(x,y) ...,xs,rep(ys,each=length(xs)))