添加一个新列,该列依赖于同一行中的其他元素

时间:2015-03-20 20:59:34

标签: wolfram-mathematica

我在Mathematica中处理我的数据集时遇到问题。基本上,我只想创建一个新列“newColumn”,它依赖于另外两列,“a”和“b”。但“a”中的值可以是x或y。如果它是x,我想将b ^ 2写入我的新列,否则写入sqrt(b)。

谢谢,非常感谢任何帮助!

一个小例子是:

列“a”:x,y,y,x 栏“b”:1,3,4,5

然后我的新列应为:1,sqrt(3),2,25

1 个答案:

答案 0 :(得分:0)

 (example = { { x, 1 } , {y, 3} , {y, 4} , {x, 5 } }) // MatrixForm

enter image description here

 (Switch[ #[[1]] , y  , Sqrt[#[[2]]] , x , #[[2]]^2] & /@ example )//MatrixForm

enter image description here