wassname/simpeg
1
0
Fork 0
mirror of https://github.com/wassname/simpeg.git synced 2026-07-04 05:19:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

make fields return vector if only one Tx

Browse Source
This commit is contained in:
rowanc1
2014-02-12 15:24:55 -08:00
parent 732ff882e0
commit c01f1a4a5a
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
simpegEM/TDEM/FieldsTDEM.py
+4 -2
View File
@@ -29,9 +29,11 @@ class FieldsTDEM(object):
self.set_e(newFields['e'], tInd)
def fieldVec(self):
u = np.ndarray((0,self.nTx))
u = np.ndarray((0, self.nTx))
for i in range(self.nTimes):
u = np.r_[u, self.get_b(i), self.get_e(i)]
if self.nTx == 1:
u = u.flatten()
return u
####################################################
@@ -64,4 +66,4 @@ class FieldsTDEM(object):
if self.e is None:
self.e = np.zeros((self.nTimes, np.sum(self.mesh.nE), self.nTx))
self.e[:] = np.nan
self.e[ind,:,:] = e
self.e[ind,:,:] = e