rust ndarrayの多次元配列に配列を代入する

前提・実現したいこと

rust ndarrayにおいて、2次元配列を定義した時、任意のindexの1次元の配列に他で定義した配列を代入したいです。

rust
1let b = array![1f64, 2, 3];
2let mut a: Array2<f64> = Array::zeros((3,3));
3a.index_axis_mut(Axis(0), 0) = b.view();

のような事をしたいです。

発生している問題・エラーメッセージ

error[E0070]: invalid left-hand side of assignment
  --> src/descriptor.rs:19:36
   |
19 |     a.index_axis_mut(Axis(0), 0) = b.view();
   |     ---------------------------- ^
   |     |
   |     cannot assign to this expression

error[E0308]: mismatched types
  --> src/descriptor.rs:19:38
   |
19 |     res.index_axis_mut(Axis(0), 0) = b.view();
   |                                      ^^^^^^^^ types differ in mutability
   |
   = note: expected struct `ArrayBase<ViewRepr<&mut f64>, _>`
              found struct `ArrayBase<ViewRepr<&f64>, _>`

エラーは型が合致していないということなのはわかるのですが、具体的な解決策が思いつかない状況です。
また、np.concatのようなメソッドがある場合はお聞きしたいです。
私がドキュメントを読んだ限りでは、そのようなものは存在していないと思いました。

行動規範の内容に同意します

回答1件

ベストアンサー

Rust ndarray の操作は、ndarray_for_numpy_users に記載の関数のリンクを追っていくと大抵見つかります。

①発生している問題の解決案
次のような感じでいかがでしょうか。
fn assign() の方は代入に ArrayBase.assign() を使用する方法を示しています。
fn assign_with_loop() の方は代入に for-loop を使用する方法を示しています。

rust
1extern crate ndarray; // 0.14.0
2use ndarray::prelude::*;
3
4fn main() {
5    let a: Array2<f64> = array![[1., 1.], [2., 2.], [3., 3.]];
6    let axis: usize = 0usize;
7    let idx: usize = 1usize;
8    let b: Array1<f64> = array![10., 10.];
9
10    assign(&mut a.clone(), &b, axis, idx);
11    assign_with_loop(&mut a.clone(), &b, axis, idx);
12}
13
14fn assign(a: &mut Array2<f64>, b: &Array1<f64>, axis: usize, idx: usize) {
15    println!("--- assign ---");
16    println!("before assigning:");
17    println!("{:?}", a);
18    a.index_axis_mut(Axis(axis), idx).assign(&b);
19    println!("after assigning {:?}", b);
20    println!(" at (axis, idx) = ({}, {}):", axis, idx);
21    println!("{:?}", a);
22}
23
24fn assign_with_loop(a: &mut Array2<f64>, b: &Array1<f64>, axis: usize, idx: usize) {
25    println!("--- assign_with_loop ---");
26    println!("before assigning:");
27    println!("{:?}", a);
28    for (v1, v2) in a.index_axis_mut(Axis(axis), idx).iter_mut().zip(b.iter()) {
29        *v1 = *v2;
30    }
31    println!("after assigning {:?}", b);
32    println!(" at (axis, idx) = ({}, {}):", axis, idx);
33    println!("{:?}", a);
34}
35