我有以下代码:
from math import *
from scipy.special import *
import matplotlib.pyplot as plt
import csv
## Definition der Parameter für Druckgleichung nach Rudnicki (1986) ##
q = 6.0/1000 # Fluidmenge pro Fläche und Zeit [l/s]
rhof = 1000 # Dichte Flüssigkeit [kg/m³]
lameu = 11.2*10**9 # Lamé-Parameter, undrained [Pa]
lame = 8.4*10**9 # Lamé-Parameter, drained [Pa]
pi # durch Pythonmodul "math" gegeben
alpha = 0.65 # Biot-Willis-Koeffizient
G = 8.4*10**9 # Schermodul [Pa]
k = 1.0e-15 # Permeabilität [m²] bzw. [Darcy]
eta = 0.001 # Viskosität des Fluids [Pa*s]
t = 1000*24*3600 # Zeit in [s]
## Beziehungen der Parameter untereinander ##
kappa = k/eta # Berechnung der Permeabilität nach Rudnicki (1986), [m³*s/kg]
print "kappa ist:",kappa # Ausgabe Permabilität
c = (kappa*(lameu-lame)*(lame+2*G))/((alpha**2)*(lameu+2*G)) # Berechnung der Diffusivität
print "c ist:",c # Ausgabe der Diffusivität
## Bereiche der Achsen in [m] ##
#def drange(start, stop, step):
# r = start
# while r <= stop:
# yield r
# r += step
xmin = 1
xmax = 20
ymin = 1
ymax = 20
## Druckberechnung um Bohrung ##
for x in range (xmin,xmax,1):
for y in range (ymin,ymax,1):
r = sqrt(x**2+y**2)
P = (q/(rhof*4*pi*kappa))*(expn(1,r**2/(4*c*t)))
z = P/1e6
#print x, " ", y, " ", z
对于我的输出,我想在csv文件中写出x,y和z的值。它看起来应该像单行内单元格中的每个x值和同一列内单元格中的每个y值,以便可以为每个x / y组合显示一个z值。它应该是这样的:
x x1 ...
y z(x,y)z(x1,y)......
y1 z(x,y1)z(x1,y1)...
...... ......
我是csv模块的新手并没有真正习惯它,特别是如何在正确的位置设置值以及何时使用它(例如在循环内部或之后)。这可能吗?或者它只能看起来像这样(类似于我的outprmented打印):
x y z(x,y)
x y1 z(x,y1)
x y2 z(x,y2) ......
很抱歉这个非常奇怪和丑陋的格式,但我认为你明白我的想法:)
任何帮助都会 - 一如既往 - 非常感谢!
答案 0 :(得分:1)
使用csv模块,您通常必须一次写入整行。我相信这段代码(替换你的最终for循环)将打印出你想要的输出:
def your_function(x,y):
r = sqrt(x**2+y**2)
P = (q/(rhof*4*pi*kappa))*(expn(1,r**2/(4*c*t)))
z = P/1e6
x_range = range(xmin,xmax,1)
y_range = range(ymin,ymax,1)
print(x_range)
for y in y_range:
this_row = [y] + [your_function(x,y) for x in x_range]
print(this_row)
如果这样做符合您的要求,则应该直接用CSV写入来替换打印功能。