Appending to the rows of a table

前端 未结 7 1824
南方客
南方客 2020-12-05 08:25

I have a two dimensional list and a one dimensional list. I would like to insert the 1D list into the 2D list as an additional column. For example:

array = {         


        
相关标签:
7条回答
  • 2020-12-05 08:52

    Another possibility is

    result = ConstantArray[0, Dimensions[array] + {0, 1}];
    result[[All, 1 ;; Last[Dimensions[array]]]] = array;
    result[[All, -1]] = column;
    

    which seems to be faster on my computer for large numeric matrices, although it requires an additional variable. If you're dealing with real-valued entries you'll want to use

    result = ConstantArray[0.0, Dimensions[array] + {0, 1}];
    

    to keep the speed gains.

    There's also

    MapThread[Append, {array, column}]
    

    which is also fast (and elegant IMO) but will unpack the result. (But if you have symbolic entries as in the example, that's not a concern.)

    0 讨论(0)
  • 2020-12-05 08:56

    Still:

    k= Partition[Flatten@Transpose[{#, {x, y, z}}], 4]&
    
    k@ {{a, 1, 2}, {b, 2, 3}, {c, 3, 4}}
    
    (*
    -> {{a, 1, 2, x}, {b, 2, 3, y}, {c, 3, 4, z}}
    *)
    
    0 讨论(0)
  • 2020-12-05 08:59

    I (sometimes) like to transpose with Flatten, as it works with a 'ragged' array.

    Map[Flatten, Flatten[{array, column}, {2}]]
    

    giving

    {{a, 1, 2, x}, {b, 2, 3, y}, {c, 3, 4, z}}

    But if, say, the column has only 2 elements

    column2 = {x, y};
    Map[Flatten, Flatten[{array, column2}, {2}]]
    

    giving

    {{a, 1, 2, x}, {b, 2, 3, y}, {c, 3, 4}}

    (Transpose will not work here)

    0 讨论(0)
  • 2020-12-05 09:00

    Though not as practical or efficient as some of the extant methods, here are two more to add to the list:

    ArrayPad[array, {0,{0,1}}, List /@ column]
    
    PadRight[array, Dimensions[array] + {0, 1}, List /@ column]
    
    0 讨论(0)
  • 2020-12-05 09:07

    How about this?

    pos = 4;
    MapThread[Insert[#1, #2, pos] &, {array, column}]
    
    0 讨论(0)
  • 2020-12-05 09:12

    For example:

     Transpose@Append[Transpose@array, column]
    

    You can also make is a function like so:

     subListAppend = Transpose@Append[Transpose@#1, #2] &;
     subListAppend[array, column]
    

    which makes it easier if you have to use it frequently. And of course if you want to insert at any place other than just the end you can use Insert[].

    subListInsert = Transpose@Insert[Transpose@#1, #2, #3] &;
    subListInsert[array, column, 2]
    --> {{a, x, 1, 2}, {b, y, 2, 3}, {c, z, 3, 4}}
    

    EDIT: Since the obligatory speed optimization discussion has started, here are some results using this and a 10000x200 array:

    ArrayFlatten@{{array, List /@ column}}:             0.020 s
    Transpose@Append[Transpose@array, column]:          0.067 s
    MapThread[Append, {array, column}]:                 0.083 s  
    MapThread[Insert[#1, #2, 4] &, {array, column}]:    0.095 s
    Map[Flatten, Flatten[{array, column}, {2}]]:        0.26 s
    ConstantArray based solution:                       0.29 s
    Partition[Flatten@Transpose[{array, column}], 4]:   0.48 s
    

    And the winner is ArrayFlatten!

    0 讨论(0)
提交回复
热议问题