在csv文件中为for循环后写入值

时间:2014-12-19 00:51:02

标签: python csv for-loop export-to-excel export-to-csv

我有以下代码:

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) ......

很抱歉这个非常奇怪和丑陋的格式,但我认为你明白我的想法:)

任何帮助都会 - 一如既往 - 非常感谢!

1 个答案:

答案 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写入来替换打印功能。