本文實例為大家分享了python繪制直線的具體代碼，供大家參考，具體內(nèi)容如下

#!/usr/bin/env python

import vtk

# 繪制通用方法

def myshow(linepolydata):

# Now we'll look at it.

lineMapper = vtk.vtkPolyDataMapper()

if vtk.VTK_MAJOR_VERSION <= 5:

lineMapper.SetInput(linepolydata)

else:

lineMapper.SetInputData(linepolydata)

lineMapper.SetScalarRange(0, 2)

lineActor = vtk.vtkActor()

lineActor.SetMapper(lineMapper)

# The usual rendering stuff.

camera = vtk.vtkCamera()

camera.SetPosition(1, 1, 1)

camera.SetFocalPoint(0, 0, 0)

renderer = vtk.vtkRenderer()

renWin = vtk.vtkRenderWindow()

renWin.AddRenderer(renderer)

iren = vtk.vtkRenderWindowInteractor()

iren.SetRenderWindow(renWin)

renderer.AddActor(lineActor)

renderer.SetActiveCamera(camera)

renderer.ResetCamera()

renderer.SetBackground(0, 0, 0)

renWin.SetSize(300, 300)

# interact with data

renWin.Render()

iren.Start()

del lineMapper

del lineActor

del camera

del renderer

del renWin

del iren

def main():

# 直線在三維坐標(biāo)系中的2個頂點

x = [(0.0, 0.0, 0.0),(1.0, 0.0, 0.0), (0.0, 1.0, 0.0)]

# We'll create the building blocks of polydata including data attributes.

linepoly = vtk.vtkPolyData()

points = vtk.vtkPoints()

lines = vtk.vtkCellArray()

scalars = vtk.vtkFloatArray()

for i in range(3):

points.InsertNextPoint(x[i])

linepoly.SetPoints(points)

line0 = vtk.vtkLine()

line0.GetPointIds().SetId(0, 0); # 第二個0表示pts中的origin點

line0.GetPointIds().SetId(1, 1); # 第二個1表示pts中的p0點

line1 = vtk.vtkLine()

line1.GetPointIds().SetId(0, 0);

line1.GetPointIds().SetId(1, 2);

lines.InsertNextCell(line0)

lines.InsertNextCell(line1)

linepoly.SetLines(lines);

colors = vtk.vtkUnsignedCharArray()

colors.SetNumberOfComponents(3);

red = [255, 0, 0]

colors.InsertNextTypedTuple(red);

green = [0, 255, 0]

colors.InsertNextTypedTuple(green);

linepoly.GetCellData().SetScalars(colors);

del points

del lines

del scalars

del colors

myshow(linepoly)

# Clean up

del linepoly

main()

以上就是本文的全部內(nèi)容，希望對大家的學(xué)習(xí)有所幫助，也希望大家多多支持萬仟網(wǎng)。

希望與廣大網(wǎng)友互動？？

點此進(jìn)行留言吧！

總結(jié)

以上是生活随笔為你收集整理的用python绘制一条直线_python绘制直线的方法的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。