52 lines
5.7 KiB
Plaintext
52 lines
5.7 KiB
Plaintext
|
<EFBFBD><EFBFBD>i m p o r t n u m p y a s n p
|
|||
|
|
i m p o r t p l o t l y . g r a p h _ o b j e c t s a s g o
|
|||
|
|
f r o m m a t h i m p o r t *
|
|||
|
|
i m p o r t m p m a t h
|
|||
|
|
f r o m i p y w i d g e t s i m p o r t i n t e r a c t , w i d g e t s , T e x t , L a y o u t
|
|||
|
|
|
|||
|
|
f o r m u l a = T e x t ( v a l u e = ' ( 1 - c o s ( ( ( 4 ) / 2 ) * x ) ) / 2 ' , l a y o u t = L a y o u t ( w i d t h = ' 1 0 0 % ' ) , d e s c r i p t i o n = ' <00>&<26><15>&A<16>&N<14>&O<16>&% <1D>&<26>$<24>&<26><><EFBFBD>&<26>%<25>&<26>%<25>&<26>%<25>&<26>%<25>&<26>%<25>&<26>%<25>&<26><><EFBFBD>&<26>$<24>&% <1D>&O<16>&N<14>&A<16>&<26><15>&' )
|
|||
|
|
N _ s l i d e r = w i d g e t s . I n t S l i d e r ( m i n = 1 , m a x = 1 6 , v a l u e = 8 , l a y o u t = L a y o u t ( w i d t h = ' 1 0 0 % ' ) , r e a d o u t _ f o r m a t = ' . 2 5 6 f ' , d e s c r i p t i o n = ' <00>&% <1D>&<26><15><15>&<26><15>&<26>$<24>&<26><07>&<26>%<25>&% <1D>&<26><15><15>&<26>&<26>&O<16>&N<14>&N <00>&<26>%<25>&$'<27>&N <00>&<26><><EFBFBD>&<26>$<24>&km<14>&<26>%<25>&<26>%<25>&<26>%<25>&<26>%<25>&<26>%<25>&<26>%<25>&km<14>&<26>$<24>&<26><><EFBFBD>&N <00>&$'<27>&<26>%<25>&N <00>&N<14>&O<16>&<26>&<26>&<26><15><15>&% <1D>&<26>%<25>&<26><07>&<26>$<24>&<26><15>&<26><15><15>&% <1D>&' )
|
|||
|
|
R A N G E _ F R O M _ S L I D E R = w i d g e t s . F l o a t S l i d e r ( m i n = 0 , m a x = 4 * ( 4 * a t a n ( 1 ) ) , v a l u e = 0 * ( 4 * a t a n ( 1 ) ) , s t e p = ( 4 * a t a n ( 1 ) ) / 4 , l a y o u t = L a y o u t ( w i d t h = ' 1 0 0 % ' ) , r e a d o u t _ f o r m a t = ' . 2 5 6 f ' , d e s c r i p t i o n = ' <00>&O<16>&<26>$<24>&% <1D>&<26><><EFBFBD>&<26>%<25>&<26><15><15>&df<15>&N <00>&<26><15>&% <1D>&<26>%<25>&<26>%<25>&<26>%<25>&<26>%<25>&<26>%<25>&<26>%<25>&% <1D>&<26><15>&N <00>&df<15>&<26><15><15>&<26>%<25>&<26><><EFBFBD>&% <1D>&<26>$<24>&O<16>&' )
|
|||
|
|
R A N G E _ T O _ S L I D E R = w i d g e t s . F l o a t S l i d e r ( m i n = 0 , m a x = 4 * ( 4 * a t a n ( 1 ) ) , v a l u e = 4 * ( 4 * a t a n ( 1 ) ) , s t e p = ( 4 * a t a n ( 1 ) ) / 4 , l a y o u t = L a y o u t ( w i d t h = ' 1 0 0 % ' ) , r e a d o u t _ f o r m a t = ' . 2 5 6 f ' , d e s c r i p t i o n = ' <00>&<26>$<24>&$'<27>&<26>%<25>&<26><15><15>&df<15>&N <00>&<26><15>&% <1D>&<26>%<25>&<26>%<25>&<26>%<25>&<26>%<25>&<26>%<25>&<26>%<25>&% <1D>&<26><15>&N <00>&df<15>&<26><15><15>&<26>%<25>&$'<27>&<26>$<24>&' )
|
|||
|
|
|
|||
|
|
d e f c l a m p ( x ) :
|
|||
|
|
r e t u r n m a x ( m i n ( 1 , x ) , - 1 )
|
|||
|
|
d e f k a p p a ( f o r m u l a , x ) :
|
|||
|
|
f u n c _ d i c t = { f n : e v a l ( f ' l a m b d a * a r g s : m p m a t h . { f n } ( * a r g s ) ' ) f o r f n i n d i r ( m p m a t h ) }
|
|||
|
|
r e t u r n f l o a t ( e v a l ( f o r m u l a , { ' x ' : x , ' c l a m p ' : c l a m p , * * f u n c _ d i c t } ) )
|
|||
|
|
d e f p l o t ( f o r m u l a = ' ( 1 - c o s ( ( ( 4 ) / 2 ) * x ) ) / 2 ' , R A N G E _ F R O M = 0 * ( 4 * a t a n ( 1 ) ) , R A N G E _ T O = 4 * ( 4 * a t a n ( 1 ) ) , N = 8 ) :
|
|||
|
|
n u m _ p o i n t s = 1 + 2 * * N
|
|||
|
|
|
|||
|
|
# G e n e r a t e x v a l u e s w i t h t h e s p e c i f i e d n u m b e r o f p o i n t s
|
|||
|
|
x _ v a l s = n p . l i n s p a c e ( R A N G E _ F R O M , R A N G E _ T O , n u m _ p o i n t s )
|
|||
|
|
|
|||
|
|
# C o m p u t e k a p p a v a l u e s
|
|||
|
|
k a p p a _ v a l s = n p . a r r a y ( [ k a p p a ( f o r m u l a , x _ v a l ) f o r x _ v a l i n x _ v a l s ] )
|
|||
|
|
|
|||
|
|
t h e t a _ v a l s = n p . c u m s u m ( k a p p a _ v a l s ) * ( x _ v a l s [ 1 ] - x _ v a l s [ 0 ] ) i f n u m _ p o i n t s > 1 e l s e n p . a r r a y ( [ 0 ] )
|
|||
|
|
x _ c o o r d s _ = n p . c u m s u m ( n p . c o s ( t h e t a _ v a l s ) ) * ( x _ v a l s [ 1 ] - x _ v a l s [ 0 ] ) i f n u m _ p o i n t s > 1 e l s e n p . a r r a y ( [ 0 ] )
|
|||
|
|
y _ c o o r d s _ = n p . c u m s u m ( n p . s i n ( t h e t a _ v a l s ) ) * ( x _ v a l s [ 1 ] - x _ v a l s [ 0 ] ) i f n u m _ p o i n t s > 1 e l s e n p . a r r a y ( [ 0 ] )
|
|||
|
|
|
|||
|
|
# C h e c k i f t h e f i r s t p o i n t i s z e r o , i f n o t , a d d i t m a n u a l l y
|
|||
|
|
i f x _ c o o r d s _ [ 0 ] ! = 0 o r y _ c o o r d s _ [ 0 ] ! = 0 :
|
|||
|
|
x _ c o o r d s = n p . i n s e r t ( x _ c o o r d s _ , 0 , 0 )
|
|||
|
|
y _ c o o r d s = n p . i n s e r t ( y _ c o o r d s _ , 0 , 0 )
|
|||
|
|
e l s e :
|
|||
|
|
x _ c o o r d s = x _ c o o r d s _
|
|||
|
|
y _ c o o r d s = y _ c o o r d s _
|
|||
|
|
|
|||
|
|
f i g = g o . F i g u r e ( )
|
|||
|
|
|
|||
|
|
f i g . a d d _ t r a c e ( g o . S c a t t e r ( x = x _ c o o r d s , y = y _ c o o r d s , m o d e = ' l i n e s ' , l i n e = d i c t ( c o l o r = ' # C E C E C E ' ) , n a m e = ' ' , h o v e r t e m p l a t e = ' X : % { x : . 2 5 6 f } ' + ' < b r > Y : % { y : . 2 5 6 f } ' ) )
|
|||
|
|
|
|||
|
|
f i g . u p d a t e _ l a y o u t (
|
|||
|
|
a u t o s i z e = T r u e ,
|
|||
|
|
x a x i s = d i c t ( s c a l e a n c h o r = ' y ' , s c a l e r a t i o = 1 , g r i d c o l o r = ' # C E C E C E ' , z e r o l i n e = T r u e , z e r o l i n e c o l o r = ' # C E C E C E ' , t i c k f o n t = d i c t ( c o l o r = ' # 9 C 9 C 9 C ' ) ) ,
|
|||
|
|
y a x i s = d i c t ( g r i d c o l o r = ' # C E C E C E ' , z e r o l i n e = T r u e , z e r o l i n e c o l o r = ' # C E C E C E ' , t i c k f o n t = d i c t ( c o l o r = ' # 9 C 9 C 9 C ' ) ) ,
|
|||
|
|
h o v e r l a b e l = d i c t ( b g c o l o r = " # F F F F F F " , f o n t _ c o l o r = ' # 9 C 9 C 9 C ' , b o r d e r c o l o r = " # C E C E C E " ) , p l o t _ b g c o l o r = ' # F F F F F F '
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
f i g . s h o w ( )
|
|||
|
|
|
|||
|
|
# C r e a t e t h e i n t e r a c t i v e p l o t
|
|||
|
|
i n t e r a c t ( p l o t , f o r m u l a = f o r m u l a , R A N G E _ F R O M = R A N G E _ F R O M _ S L I D E R , R A N G E _ T O = R A N G E _ T O _ S L I D E R , N = N _ s l i d e r ) ;
|